目標檢測---資料集格式轉化及訓練集和驗證集劃分_didiaopao的博客-CSDN博客
1、 我們經常從網上獲取一些目標檢測的資料集資源標簽的格式都是VOC(xml格式)的,而yolov5訓練所需要的檔案格式是yolo(txt格式)的,這里就需要對xml格式的標簽檔案轉換為txt檔案,同時訓練自己的yolov5檢測模型的時候,資料集需要劃分為訓練集和驗證集,這里提供了一份代碼將xml格式的標注檔案轉換為txt格式的標注檔案,并按比例劃分為訓練集和驗證集,先上代碼再講解代碼的注意事項
目標檢測---資料集格式轉化及訓練集和驗證集劃分_didiaopao的博客-CSDN博客
2、標簽為yolo格式資料集劃分訓練集和驗證集
由于yolov5訓練需要的資料標簽格式為txt格式,所以大家在利用labelimg標注的時候會用yolo格式(標注生成的標簽為txt格式),標注好的資料集訓練的時候就要劃分為訓練集和驗證集,因此就需要有劃分為訓練集和測驗集的代碼,這里需要講的是我寫的腳本代碼可以成功將資料集劃分為訓練集和驗證集,但是在訓練模型的時候,加載資料集一直會出現問題,因此我就想到了,先把txt格式的資料集替換成xml格式的資料集,然后再按上述將xml格式標簽轉化為txt格式標簽并劃分為訓練集和驗證集的方法劃分就好了,但是這里建議大家以后標注的時候就標注為voc格式(xml格式),因為該格式的標簽里面有圖片標注的具體內容,例如標注類別,圖片大小,標注坐標,但是yolo格式(txt格式)里面是用數字來代表類別,這樣很不直觀,而且標注的坐標也是經過轉化歸一化的,坐標資訊更加不直觀,先上yolo轉voc的代碼,
3、 訓練自己的資料集
目標檢測---教你利用yolov5訓練自己的目標檢測模型_didiaopao的博客-CSDN博客
4、學習網站
深入淺出Yolo系列之Yolov5核心基礎知識完整講解_江南研習社-CSDN博客_yolov5的輸出
【Yolov5】1.爆肝6000字Yolov5保姆級教程,80歲老奶奶都看得懂!_若丶塵的博客-CSDN博客

轉載請註明出處,本文鏈接:https://www.uj5u.com/qita/336329.html
標籤:其他
上一篇:成語接龍敢玩嘛:—偷偷學成語,然后驚艷所有人!龍騰虎躍,該你了?【評論區見】
下一篇:2021-10-25
