yolov3 修改 只識別一類(person)

2021-08-25 11:49:18 字數 541 閱讀 6139

yolov3可以識別多種目標,但是我們可能只需要識別一類物體,例如person。

在編譯yolov3之後,預設可以識別多種目標

之前在網上看其他人修改yolov2版本,可以只識別person

但是yolov3版本的**有變化,需要做出其他修改。在網上找了一下,沒有發現yolov3的相關修改,於是自己修改了一下

修改檔案:darknet-master\src\image.c

修改函式為:draw_detections_v3

在287行:const int best_class = selected_detections[i].best_class;後插入:

if(strcmp(names[best_class], "person") !=0 )

在306行:for (i = 0; i < selected_detections_num; ++i) 然後重新編譯即可

相當於在演算法識別所有類別後只輸出person類。當然也可以自己重新訓練,使之只識別person

參考部落格:

yolov3系列 零 yolov3詳解

目標檢測演算法與efficientdet講解 論 文 翻譯 yolov3主頁 yolo系列的目標檢測演算法可以說是目標檢測史上的巨集篇巨作,v3演算法是在v1 v2基礎上形成的,先看下 yolov1 yolov2 下圖為yolov3的網路結構 dbl darknetconv2d bn leaky 是...

YOLOv3 從入門到部署 (一)YOLOv3概述

這是目標檢測領域常用的兩個資料集。voc和coco的區別主要在於影象標註的格式不一樣。voc將目標的標籤資訊以某種格式儲存在.xml格式的檔案中,而coco則是儲存在.txt檔案中。因此有時候voc和coco不是指資料集,而是指資料集標註格式。我們完全沒有必要去研究每種標註格式的細節,網路也有大量的...

yolov3的缺點 yolov3特點總結

p1,yolov3,簡單總結下比較特別的地方 1.bounding box 中心值x,y的 是通過乙個sigmoid 函式,來 在每乙個單元 cell 的相對位置。而不像普通那種 乙個相對於anchor的offset。然後bbox 損失是用的mse 一般都是用smooth l1 3.類別 沒有使用s...