如何在自定義的數(shù)據(jù)集上訓練YOLOv5?
選擇“YOLO v5 Pythorch”當出現(xiàn)提示時,一定要選擇“Show Code Snippet”,這將輸出一個下載curl腳本,這樣你就可以輕松地將數(shù)據(jù)以正確的格式移植到Colab中。curl -L "https://public.roboflow.a(chǎn)i/ds/YOUR-LINK-HERE" > roboflow.zip; unzip roboflow.zip; rm roboflow.zip正在Colab中下載…
下載YOLOv5格式的自定義對象數(shù)據(jù)集導出將會創(chuàng)建一個名為data.yaml的YOLOv5.yaml文件,指定YOLOv5 images文件夾、YOLOv5 labels文件夾的位置以及自定義類的信息。定義YOLOv5模型配置和架構(gòu)接下來,我們?yōu)槲覀兊亩ㄖ茖ο髾z測器編寫一個模型配置文件。在本教程中,我們選擇了最小、最快的YOLOv5基本模型,你也可以從其他YOLOv5模型中選擇,包括:YOLOv5sYOLOv5mYOLOv5lYOLOv5x你也可以在此步驟中編輯網(wǎng)絡結(jié)構(gòu),但一般不需要這樣做。以下是YOLOv5模型配置文件,我們將其命名為custom_yolov5s.yaml:nc: 3depth_multiple: 0.33width_multiple: 0.50
anchors: - [10,13, 16,30, 33,23] - [30,61, 62,45, 59,119] - [116,90, 156,198, 373,326]
backbone: [[-1, 1, Focus, [64, 3]], [-1, 1, Conv, [128, 3, 2]], [-1, 3, Bottleneck, [128]], [-1, 1, Conv, [256, 3, 2]], [-1, 9, BottleneckCSP, [256]], [-1, 1, Conv, [512, 3, 2]], [-1, 9, BottleneckCSP, [512]], [-1, 1, Conv, [1024, 3, 2]], [-1, 1, SPP, [1024, [5, 9, 13]]], [-1, 6, BottleneckCSP, [1024]], ]
head: [[-1, 3, BottleneckCSP, [1024, False]], [-1, 1, nn.Conv2d, [na * (nc + 5), 1, 1, 0]], [-2, 1, nn.Upsample, [None, 2, "nearest"]], [[-1, 6], 1, Concat, [1]], [-1, 1, Conv, [512, 1, 1]], [-1, 3, BottleneckCSP, [512, False]], [-1, 1, nn.Conv2d, [na * (nc + 5), 1, 1, 0]], [-2, 1, nn.Upsample, [None, 2, "nearest"]], [[-1, 4], 1, Concat, [1]], [-1, 1, Conv, [256, 1, 1]], [-1, 3, BottleneckCSP, [256, False]], [-1, 1, nn.Conv2d, [na * (nc + 5), 1, 1, 0]],
[[], 1, Detect, [nc, anchors]], ]訓練定制YOLOv5探測器我們的data.yaml和custom_yolov5s.yaml文件已經(jīng)準備好了,我們庫開始訓練了!為了開始訓練,我們使用以下選項運行訓練命令:img:定義輸入圖像大小batch:確定batch大小epochs:定義epochs。(注:通常,3000+很常見。ヾata:設置yaml文件的路徑cfg:指定我們的模型配置weights:指定權(quán)重的自定義路徑。(注意:你可以從Ultralytics Google Drive文件夾下載權(quán)重)name:結(jié)果名稱nosave:只保存最后的檢查點cache:緩存圖像以加快訓練速度運行訓練命令:
訓練定制的YOLOv5探測器。它訓練得很快!在訓練期間,你可以看 mAP@0.5 來了解你的探測器是如何運行的,請參閱這篇文章。https://blog.roboflow.a(chǎn)i/what-is-mean-average-precision-object-detection/評估定制YOLOv5檢測器的性能現(xiàn)在我們已經(jīng)完成了訓練,我們可以通過查看驗證指標來評估訓練過程的執(zhí)行情況。訓練腳本將刪除tensorboard日志,我們將其可視化:

請輸入評論內(nèi)容...
請輸入評論/評論長度6~500個字
最新活動更多
-
7月22-29日立即報名>> 【線下論壇】第三屆安富利汽車生態(tài)圈峰會
-
7.30-8.1火熱報名中>> 全數(shù)會2025(第六屆)機器人及智能工廠展
-
7月31日免費預約>> OFweek 2025具身智能機器人產(chǎn)業(yè)技術(shù)創(chuàng)新應用論壇
-
免費參會立即報名>> 7月30日- 8月1日 2025全數(shù)會工業(yè)芯片與傳感儀表展
-
即日-2025.8.1立即下載>> 《2024智能制造產(chǎn)業(yè)高端化、智能化、綠色化發(fā)展藍皮書》
-
8月5日立即報名>> 【在線會議】CAE優(yōu)化設計:醫(yī)療器械設計的應用案例與方案解析
推薦專題