[Tensorflow] COCO预训练模型和重新训练模型中的不同图形架构

时间:2018-02-05 08:53:38

标签: tensorflow object-detection

我已按照文档: Semantic Relatedness of Words and Phrases 重新训练ssd_mobilenet_v1_coco_2017_11_17.tar.gz的模型。  我只用修改管道配置 - 微调至真 - 设置模型路径 - 使用tfrecord的实际路径设置训练的数据路径

然后,我尝试使用导出脚本: https://github.com/tensorflow/models/blob/master/research/object_detection/g3doc/running_locally.md 最后我用自己的数据获得了pb。

但我发现重新训练模型的结果与ssd_mobilenet_v1_coco_2017_11_17.tar.gz中的结构不同。

这是我从Tensorboard拍摄的照片:

根据tensorboard的图片,我发现BoxPredictor中有两个输入张量,但有三个原始张量。

有谁可以帮助我解决这个问题? 感谢...

PS:我使用版本1.4.0-GPU的tensorflow

0 个答案:

没有答案