iT邦幫忙

2023 iThome 鐵人賽

DAY 26
0
AI & Data

YOLO系列網路技術採用以及實作系列 第 26

[DAY 26] YOLOv8模型訓練

  • 分享至 

  • xImage
  •  

建立所需環境

  1. 利用anaconda創建一個yolov8的環境,並啟動。
    conda create -n yolov8 python==3.8
    conda activate yolov8
    
  2. 安裝yolov8套件
    pip install ultralytics
    
  3. 下載yolov8程式碼
  4. 進到下載好的程式資料夾中,安裝剩餘的套件。
    pip install -e ultralytic
    
  5. 預測一下圖片,確認安裝是否完成:
    yolo task=detect mode=predict model=yolov8n.pt conf=0.25 source='./dog.jpg'
    
    https://ithelp.ithome.com.tw/upload/images/20231011/20120310rg863GDJKf.png
    *輸出圖片會存放在runs\deetect\predict資料夾中:
    https://ithelp.ithome.com.tw/upload/images/20231011/201203105Rzm3QCUVj.png
    • 執行完成這個指令的同時,會將yolov8n.pt權重檔下載下來。

訓練階段(以飛機機身缺陷偵測資料集為例)

  1. 創建一個名稱為data的資料夾,並將aircraft_fuselage_yolo的資料集內容複製進去。
  2. 在data資料夾中,再創建一個custom.yaml的檔案,存放以下內容,也可以直接將yolov7資料夾中,放在data資料夾中的custom.yaml複製過去使用:
    train: '.\data\images\train'
    val: '.\data\images\val'
    test: 
    nc: 4
    # class names
    names: ['scratch','paint_peel','rust','rivet_damage']
    
  3. 執行訓練指令:
    yolo task=detect mode=train model=yolov8n.pt imgsz=640 data=.\data\custom.yaml epochs=300 batch=2 name=yolov8_it30
    

成功執行會出現以下圖片的畫面,並開始訓練。

https://ithelp.ithome.com.tw/upload/images/20231011/20120310Tfyp4SbM1Q.png

這邊同樣只演示能成功進行訓練的步驟,沒有讓訓練跑到底,後續回到宿舍後,會利用散熱效果較好的桌上型電腦進行訓練,並在最後的幾篇文章中,統一將每一代的YOLO訓練結果數據進行展示。


上一篇
[DAY 25] YOLOv7模型訓練
下一篇
[DAY 27] 訓練資料前處理(補充)
系列文
YOLO系列網路技術採用以及實作30
圖片
  直播研討會
圖片
{{ item.channelVendor }} {{ item.webinarstarted }} |
{{ formatDate(item.duration) }}
直播中

尚未有邦友留言

立即登入留言