• pytorch-yolov5
• tensorflow-yolov5

由于流程比較完備，主要準備好數據就可以了。不過我的方法比較曲折，歷史包袱重了點

準備數據

首先，我根據tensorflow版本中常用的方法，將數據準備為一個文檔格式。
參考了這個腳本， 然后運行

data_prepare = DataPrepare(data_dir=base_dir, output_dir=base_dir + '/data') data_prepare.write()

數據轉換

首先嘗試pytorch版，需要把數據再轉一下。tensorflow版本可以直接跑了。

總結

以上是生活随笔為你收集整理的机器学习实战：训练自己的YoloV5 [草稿-待完成]的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。