iT邦幫忙

2023 iThome 鐵人賽

DAY 16
0
AI & Data

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

[DAY 16] YOLOv8

  • 分享至 

  • xImage
  •  

YOLOv8: the eighth iteration of the revolutionary "You Only Look Once" object detection model

Glenn Jocher : YOLOv5、YOLOv8作者及Ultralytics公司的創始人。

  • YOLOv8為Ultralytics在今年一月的時候出版的,很大部分修改自YOLOv5,且同YOLOv5一樣目前為止沒有出版paper。
    https://ithelp.ithome.com.tw/upload/images/20231001/201203101gDDR7JEAq.png
  • 從YOLOv8官方網站Docs說明中得到,YOLOv8主要為:
    • 採用最好的BackBone、Neck架構可以提升特徵提取以及物件偵測的表現。
    • 跟YOLOX一樣採用Anchor Free的架構。
    • 在準確度及預測速度之間要取得平衡。
    • 提供多樣化的預訓練模型用來處理各種任務,讓特定任務能夠更快的找到正確的模型。
  • YOLOv8在參數量、運算速度與準確度的關係(跟YOLOv5、YOLOv6以及YOLOv7的比較):
    https://ithelp.ithome.com.tw/upload/images/20231001/20120310jSyHZThCVT.png

架構圖

  • YOLOv8架構跟YOLOv5很像,因此我是從我YOLOv5畫的圖為基礎進行修改,YOLOv8改動的地方我有用不同顏色標示:
    https://ithelp.ithome.com.tw/upload/images/20231001/20120310KjF1WwSrzQ.png

    本架構圖為自行繪製,若有錯誤,歡迎指正。

  • 原本YOLOv5的CBL架構激活函數採用Leakly ReLu,YOLOv8改成SiLu。
  • 原本YOLOv5的CSP架構,YOLOv8可能參考YOLOv7提到的ELAN架構結合成CSP-ELAN架構。
  • 紅色的那個CBL Block被移除了。
  • 後面的Prediction Head架構,可能參考YOLOX提到的Anchor Free架構,將Bounding Box回歸跟Class預測分開計算。

損失函數

  • YOLOv8沒有Objectness Loss。
  • Class Loss:採用Binary Cross Entropy Loss。
  • Bounding Box Loss:採用C-IoU以及Distribution Focal Loss(DFL)。

其他提升準確度的方法

  • 訓練的最後10個Epoch會關閉Mosiac數據增強技術。
  • 訓練的Epoch數量由300提升到500。

文章使用之圖片擷取自該篇論文

明日將提前進入到實作環節。

參考文獻

  1. YOLOv8官方Docs網站
  2. 網路上其他人畫的架構圖
  3. YOLOv8架構解說-參考網站1
  4. YOLOv8架構解說-參考網站2
  5. YOLOv8架構解說-參考網站3
  6. YOLOv8架構解說-參考網站4

上一篇
[DAY 15] YOLOv7-下篇
下一篇
[DAY 17] 常用的公開資料集網站
系列文
YOLO系列網路技術採用以及實作30
圖片
  直播研討會
圖片
{{ item.channelVendor }} {{ item.webinarstarted }} |
{{ formatDate(item.duration) }}
直播中

1 則留言

0
阿醺
iT邦新手 5 級 ‧ 2024-05-30 15:20:23

您好,想請問如果我已經有yolov5訓練過的環境,要怎樣才能轉成使用yolov8呢?還是需要重新佈署環境?

Xian iT邦新手 5 級 ‧ 2024-06-12 13:24:44 檢舉

您提問的採用yolov5環境訓練yolov8,個人認為yolov5無法向上兼容yolov8,個人還沒嘗試過,但建議還是新部屬環境為佳。因為yolov8這邊只需下'pip install ultralytics'即可將yolov8環境安裝完成,若要採用yolov8環境訓練yolov5模型,嘗試過無法直接採用yolov8程式碼直接執行yolov5訓練程式,可參考以下連結進行修改:
https://github.com/ultralytics/ultralytics/issues/4647

我要留言

立即登入留言