微调的 Inception V3 输出张量形状

Question

我按照Tensorflow Slim存储库中提供的教程，使用花卉数据集微调InceptionV3模型。在冻结图形并在我自己的代码中使用它方面，我能够一直到教程的结尾。但是，我的印象是模型的微调，将输出张量从(?, 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