Tensorflow SSD对象检测类别选择

Question

我正在研究基本的Tensorflow对象检测示例here 它正在从coco数据集中检测所有90个类别。但是我只想从中检测两个类。该怎么做？

Answer 1

让我们说您只想检测摩托车和人。 在文件visualisation_util.py中，转到def draw_bounding_box_on_image_array ，有一个功能：

if (display_str_list[0][0:3]=="per" or display_str_list[0][0:3]=="mot"):

  draw_bounding_box_on_image(image_pil, ymin, xmin, ymax, xmax, color,
                         thickness, display_str_list,
                         use_normalized_coordinates)

将此函数调用放在这样的条件下

orbeon_form_definition

此处“ per”是“ person”的前三个字母，“ mot”是机车的前三个字母。这样，您可以从所有其他对象中检测出所需的对象