ssd_mobilenet_v2_fpnlite_320x320 的 .pbtxt 文件

.pbtxt file for ssd_mobilenet_v2_fpnlite_320x320

抱歉,我是使用 Tensorflow 2.0 制作模型的新手。

我正在尝试使用来自 Tensorflow 2.0 Object Detection Zoo 的预训练模型训练模型:ssd_mobilenet_v2_fpnlite_320x320。我正在学习使用模型的 .pbtxt 文件的教程,但我不确定如何获取它,因为模型只有一个 .pb 文件。

https://github.com/tensorflow/models/blob/master/research/object_detection/g3doc/tf2_detection_zoo.md 这里是link。我计划为我的模型使用 SSD MobileNet v2 320x320。

基本上我的问题是:如何获取 .pbtxt 文件以在 pipeline.config 文件中引用,或者我是否可以为 label_map_path 添加其他内容?

train_input_reader {

label_map_path:“路径”

tf_record_input_reader {

input_path:“路径”

}

}

您必须根据要训练 object-detection 模型的内容创建“pbtxt”文件。 pbtxt文件格式为:

item {
    id: 1
    name: 'class name 1'
}

item {
    id: 2
    name: 'class name 2'
}

例如,如果您正在训练对象检测模型来检测猫和狗。 您的 label_map.pbtxt 文件看起来像。

item {
    id: 1
    name: 'cat'
}

item {
    id: 2
    name: 'dog'
}

注意:此名称应与您在标记图像时为类别指定的名称完全匹配。即使用labelImg软件进行标注。

有关使用 Tensorflow 对象检测的完整指南API。 参见 here

创建 label_map.pbtxt 文件后,您只需在 pipeline.pbtxt 中指定文件的路径。

train_input_reader {

label_map_path: "PATH_TO_LABEL_MAP_FOLDER/label_map.pbtxt"

tf_record_input_reader {

input_path: "PATH"

}

}