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"
}
}
抱歉,我是使用 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"
}
}