[深度學習] 如何使用 YOLO 製作即時口罩檢測系統(二) – 建立口罩檢測模型?

  • 大數學堂
  • 2020年4月16日 09:17

在安裝完YOLOv3的安裝程序後,我們開始要建立我們的口罩檢測模型。

要做人工智慧前,必定先要有人工標記的資料集,因此我們先下載Kaggle 上的口罩資料集,並將資料上載至Google Drive,接者將標注好的Label XML ,轉換為YOLOv3 可以接受的輸入格式。

設定好模型所需之設定檔案(obj.data, obj.name, train.txt, test.txt, yolov3-tiny.cfg)並下載預訓練模型 darknet53.conv.74後,我們便可以開始訓練我們的口罩檢測模型了!

影片: https://largitdata.com/course/127/

程式碼:https://github.com/ywchiu/largitdata/blob/master/code/Course_127.ipynb

#DeepLearning

#YOLO

#COVID19

#新冠肺炎

#口罩檢測