尝试在张量流对象检测 api 中导出推理时出现不完整的形状错误

Question

我用 loss 2.x 训练了自定义对象检测器，当我尝试导出它时出现以下错误

我遇到了这个：'Parsing Inputs... Incomplete shape' error while exporting the inference graph in Tensorflow

但我确定标签是相同的：我多次运行它

python3 export_inference_graph.py --input_type image_tensor       --pipeline_config_path training/inception_v2.config --trained_checkpoint_prefix training/model.ckpt-688 --output_directory trained-inference-graphs/output_inference_graph_v1

我收到了错误：

由于形状不完整，114 次操作没有失败统计数据。 解析输入... 形状不完整。

完整报告请查看：https://pastebin.com/mGSBDgJC

编辑：模型使用 eval 检测 python3 eval.py --logtostderr --pipeline_config_path=/home/ic/Documents/objectExtraction/workspace/training_demo/training/inception_v2.config --checkpoint_dir=/home/ic/Documents/objectExtraction/workspace/training_demo/training --eval_dir =/home/ic/Documents/objectExtraction/workspace/training_demo/eval

Answer 1

发生这种情况是因为您没有使用参数input_shape 说明输入分辨率是多少。它只是意味着它无法计算一个操作需要多少次触发器，因为它不知道输入分辨率是多少。您仍然可以毫无问题地使用导出的图进行推理。