微调 Inception V3 output 张量形状

Question

我正在按照Tensorflow Slim存储库中提供的教程使用鲜花数据集微调InceptionV3 model。 在冻结图形并在我自己的代码中使用它方面，我能够一直到教程结束 go。 然而，我的印象是 model 的微调，将 output 张量从(?, 1001)更改为(?, 5)形状。 但形状仍然是(?, 1001) 。 我在这里错过了什么吗？

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

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

Answer 1

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

$ 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