tensorflow yolov3只检测行人

官方给的模型能够检测80多个类别,这里咱们只检测行人,就须要将代码进行稍微修改便可: 项目路径图: 查看./data/classes/coco.names 类别的索引: 这里person为第1个,即索引为0 更改:web ./core/utils.py draw_bbox函数(将roi_index设置为你想要的索引便可,我这里为0):dom def draw_bbox(image, bboxes,
相关文章
相关标签/搜索