Python影像辨識筆記(二十一):ultralytics/yolov5使用教學及注意事項

正文:

ultralytics/yolov5是由國外一間公司用PyTorch實現的YOLOv5

特點:

  1. 能自動從Google Drive下載缺少的Weight
  2. 能在不需要OpenCV(C++)的環境下進行資料增強及Model訓練
  3. 可以直接進行mAP測試,不須上傳至CodaLab,不過測試時間大概需要25分鐘(以官方weight為例)

除此之外,更多特點可以參考官方GitHub REPO:

注意事項:

ultralytics的COCO路必須放在yolov5資料夾的平行目錄。

例如:

與 ultralytics/yolov3的一些差異:

使用方式:

請注意自己的python指令預設是python2還是python3版本,如果是python3版本可以放心使用下列指令;若不是,請使用python3指令

--

--

Machine Learning | Deep Learning | https://linktr.ee/yanwei

Love podcasts or audiobooks? Learn on the go with our new app.

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