使用tensorflow时,在哪里可以找到ssd_mobilenet_v1_coco的label map file(pbtxt)?

Where can I find the label map file(pbtxt) of ssd_mobilenet_v1_coco when using tensorflow?

我在 windows 10 上用 tensorflow object detection 学习了对象检测。

我从 https://github.com/tensorflow/models/blob/master/research/object_detection/g3doc/detection_model_zoo.md

下载 ssd_mobilenet_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 相关?

label map不是特定于架构,而是特定于数据集(类你支持,因此你应该设置分类层的通道数)。 因此,你只需要coco对应的label map,即object_detection/data/mscoco_label_map.pbtxt.

模型中实现了数据集。模型通过 tar.gz 或 .zip 交付。如果你使用预训练的基本模型,那么标签图可以在代码树 githup object_detection/data/mscoco_label_map.pbtxt 中找到,正如 netanel-sam 解释的那样。

但是如果你开始训练你的预训练模型并将要检测的项目添加到你的数据集中并开始交付你修改后的模型,那么你必须也提供你的标签地图并且没有比将它包含到.tar.gz 或 .zip。同样的情况也发生在精简模型上,因为从可训练模型到精简模型的转换通常会丢失数据集中的项目。与基本模型相比,精简模型还使用其他格式来标记映射。令人困惑?