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,
相關文章
相關標籤/搜索