選擇“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模型配置和架構接下來，我們?yōu)槲覀兊亩ㄖ茖ο髾z測器編寫一個模型配置文件。在本教程中，我們選擇了最小、最快的YOLOv5基本模型，你也可以從其他YOLOv5模型中選擇，包括：YOLOv5sYOLOv5mYOLOv5lYOLOv5x你也可以在此步驟中編輯網(wǎng)絡結(jié)構，但一般不需要這樣做。以下是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：指定權重的自定義路徑。（注意：你可以從Ultralytics Google Drive文件夾下載權重）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日志，我們將其可視化：