iT邦幫忙

0

YOLOv4只顯示一類檢測結果

各位好,目前是初學者,遇到了以下的問題:

在訓練 YOLOv4模型中,訓練了汽車、卡車、公車、摩托車,想問目前只想採用汽車的bounding box作為輸出,請問有辦法嗎?還是需要重新訓練?

圖片
  直播研討會
圖片
{{ item.channelVendor }} {{ item.webinarstarted }} |
{{ formatDate(item.duration) }}
直播中

1 個回答

1
wrxue
iT邦好手 1 級 ‧ 2021-07-01 21:01:44
最佳解答

單純使用 darknet detector test 無法達到您想要的效果,但這不意味著需要重新訓練,只需要另外寫 Python 程式,可參考我之前做的 YOLOv4 應用範例

全民來找碴 - YOLOv4 戴口罩檢測 # 將 YOLOv4 整合在應用程式中的核心程式碼

其中 r 是您的 YOLOv4 偵測出來的 Bounding Boxes,只需在 draw_boxes 之前過濾掉您不想顯示的類別即可

r = darknet.detect_image(network, class_names, img)
...
...
# 在這加上過濾的程式碼,把非汽車的 Bounding Boxes 都去除
img = darknet.draw_boxes(r, img, colors)

我要發表回答

立即登入回答