tensorflow - 在 Tensorflow 中导出推理图时出现“解析输入...形状不完整”错误

Question

我正在使用 Tensorflow 的对象检测 API 训练神经网络来检测汽车。我使用以下 senddex 的 YouTube 视频来学习和执行该过程。

https://www.youtube.com/watch?v=srPndLNMMpk&t=65s

还有他的视频的文字版本:

https://pythonprogramming.net/testing-custom-object-detector-tensorflow-object-detection-api-tutorial/?completed=/training-custom-objects-tensorflow-object-detection-api-tutorial/

他的系列的第 1 至 6 部分。

我的训练数据有约 300 张图像，测试数据有约 20 张图像

由于我使用的是迁移学习的概念，因此我选择了 ssd_mobilenet_v1.model。

Tensorflow 版本 1.10(在 CPU 上)Python 版本 3.6

我训练了约 8000 步，损失了约 1 步。

一切都运行顺利，直到我尝试导出推理图。

我运行以下命令:

python3 export_inference_graph.py \
--input_type image_tensor \
--pipeline_config_path training/ssd_mobilenet_v1_pets.config \
--trained_checkpoint_prefix training/model.ckpt-7882 \
--output_directory car_inference_graph

执行此命令后，我得到

警告:

WARNING:tensorflow:From 
/home/mydir/Desktop/models/research/object_detection/exporter.py:280: 
get_or_create_global_step (from 
tensorflow.contrib.framework.python.ops.variables) is deprecated and will be 
removed in a future version.
Instructions for updating:
Please switch to tf.train.get_or_create_global_step
WARNING:tensorflow:From 
/home/mydir/Desktop/models/research/object_detection/exporter.py:434: 
print_model_analysis (from tensorflow.contrib.tfprof.model_analyzer) is 
deprecated and will be removed after 2018-01-01.
Instructions for updating:
Use `tf.profiler.profile(graph, run_meta, op_log, cmd, options)`. Build 
`options` with `tf.profiler.ProfileOptionBuilder`. See README.md for details
106 ops no flops stats due to incomplete shapes.
Parsing Inputs...
Incomplete shape.

然后是模型报告列表。 (需要的话我可以发给你)

我有导出的 car_inference_graph 文件夹，其中包含检查点文件等。就像它应该有的那样在里面。

但是当我在汽车图像(测试图像或新图像)上运行此模型时，它不会检测到汽车。即没有边界框!

我不确定这个警告是否影响了检测，如果影响了，请帮我解决!

谢谢!

PS:为了帮助确定“无边界框”的原因，值得一提的是，它甚至没有在模型训练的图像中显示汽车周围的边界框。这意味着推理图的导出或边界框函数本身有问题。

我刚刚发现“.pbtxt”文件的名称为:macncheese，而不是名称:car。是它造成了这个问题吗？

Answer 1

标记期间使用的标签名称和 pbtxt 条目应该完全匹配。在您的情况下，由于名称不同，您必须更改 pbtxt 文件中的名称并重新运行训练。