【发布时间】:2019-08-05 19:30:17
【问题描述】:
我正在使用 Tensorflow 对象检测 API 项目 (https://github.com/tensorflow/models/tree/r1.12.0/research/object_detection) 处理 Mask R CNN 模型。我坚持使用 r1.12.0 版本(但这不是必须的,但我认为它不会影响我的问题。)我的计划是修改模型的一些“静态”部分并再次将其导出为冻结图格式。
作为第一步,我打算使用 export_inference_graph.py 脚本 (https://github.com/tensorflow/models/blob/r1.12.0/research/object_detection/export_inference_graph.py) 从检查点文件和 pipeline.config 重新生成冻结图。我下载了 inception V2 模型 (http://download.tensorflow.org/models/object_detection/mask_rcnn_inception_v2_coco_2018_01_28.tar.gz) 并使用 Tensorflow 1.12.0 执行了脚本。它完成了这项工作并创建了一个冻结图。
问题是,如果我将原始冻结图与生成的图进行比较,它们是不同的。如果我使用 Tensorboard 将它们可视化,它们之间就会有明显的差异。一些节点丢失,一些节点不同等等。
我也尝试过其他模型(普通 Fast R CNN),我总是遇到同样的问题。
这怎么可能?我应该如何使用检查点文件和pipeline.config 文件重新生成与最初附加的完全相同的冻结图?
【问题讨论】:
标签: python tensorflow object-detection-api