python - 微调 Inception V3 输出张量形状

Question

我按照Tensorflow SlimInceptionV3存储库中提供的教程使用鲜花数据集微调模型。在冻结图形并在我自己的代码中使用它方面，我能够一直走到教程的结尾。(?, 1001)然而，我的印象是通过将输出张量从(?, 5)形状更改为模型的微调。但形状依旧(?, 1001)。我在这里错过了什么吗？

我正在用鲜花数据集对其进行测试，但最终我的目标是训练一个模型，其中包含 ImageNet 的 1001 中不存在的类。

任何帮助表示赞赏。谢谢。

Answer 1

官方文档没有提供导出推理图所需的所有参数。切换自

$ python export_inference_graph.py \
  --alsologtostderr \
  --model_name=inception_v3 \
  --output_file=/tmp/inception_v3_inf_graph.pb

至

python3 export_inference_graph.py \
    --alsologtostderr \
    --model_name=inception_v3 \
    --train_dir=training/ \
    --dataset_dir=/tmp/flowers/ \
    --dataset_name=flowers \
    --output_file=/tmp/inception_v3_inf_graph.pb