Python影像辨識筆記(二十六):如何透過YOLOv3/v4進行自動化Bounding Box標記?

實作的思路

在上文當中介紹了如何進行自動化標記圖片,我將其流程統整如下:

  1. 在OpenCV載入Pre-trained的YOLO weight
  2. 透過這個weight,自動偵測圖片,取得預測Bounding box的座標數值
  3. 將其數值寫入到xml文件當中
  4. 使用labellmg軟體進行Bounding box數值微調
  5. 若要使用coco格式的訓練方式,可以用 convert2Yolo的程式,將voc格式的xml檔案轉換成txt格式
  6. 開始訓練自己的模型

程式碼

Get the Medium app

A button that says 'Download on the App Store', and if clicked it will lead you to the iOS App store
A button that says 'Get it on, Google Play', and if clicked it will lead you to the Google Play store