YOLOv4只顯示一類檢測結果

yolo yolov4

h402272000 2021-07-01 15:06:03 ‧ 2217 瀏覽

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

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

直播研討會

{{ item.channelVendor }} {{ item.webinarstarted }} |

直播中

wrxue

iT邦好手 1 級 ‧ 2021-07-01 21:01:44

最佳解答

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

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

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

立即登入回答