我在windows10的tensorflow object detection上学习物体检测。
我从https://github.com/tensorflow/models/blob/master/research/object_detection/g3doc/detection_model_zoo.md下载了ssd_移动网络v1_coco_2018_01_28.tar.gz
解压缩ssd_mobilenet_v1_coco_2018_01_28.tar.gz文件后，我没有找到pbtxt文件。
在哪里可以找到ssd_mobilenet_v1_coco的相关pbtxt文件？
我知道models-master\research\object_detection\data文件夹中有一些pbtxt文件，但哪个文件与ssd_mobilenet_v1_coco相关？
2条答案
按热度按时间kwvwclae1#
标签Map不是特定于架构的，而是特定于数据集的（您支持哪些类，并相应地设置分类层的通道数），因此，您只需要与coco对应的标签Map，即
object_detection/data/mscoco_label_map.pbtxt。cu6pst1q2#
数据集在模型中实现，模型以tar.gz或. zip的形式发布，如果使用预先训练好的基本模型，那么标签Map可以通过代码树githup object_detection/data/mscoco_label_map.pbtxt找到，如netanel-sam所述。
但是，如果您开始训练预训练的模型，向数据集中添加要检测的项目，并开始交付修改后的模型，那么您还必须提供标签Map，没有比将其包含到. tar.gz或. zip中更好的方法了。因为从可训练模型到精简模型的转换经常会丢失数据集中的项目。精简模型也使用其他格式来标记Map而不是基本模型。