我已按照文档: 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拍摄的照片:
重新训练我自己的数据和管道配置 https://github.com/tensorflow/models/blob/master/research/object_detection/g3doc/exporting_models.md
ssd_mobilenet_v1_coco_2017_11_17.tar.gz中来自pb文件的原始文件 https://i.stack.imgur.com/kegcT.jpg
根据tensorboard的图片,我发现BoxPredictor中有两个输入张量,但有三个原始张量。
有谁可以帮助我解决这个问题? 感谢...
PS:我使用版本1.4.0-GPU的tensorflow