微调 Inception V3 输出张量形状答案

【问题标题】：Fine-tuned Inception V3 output tensor shape微调 Inception V3 输出张量形状
【发布时间】：2020-05-14 15:45:43
【问题描述】：

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

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

感谢任何帮助。谢谢。

【问题讨论】：

标签： python tensorflow tensorflow-slim

【解决方案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

【讨论】：