參考 ：https://gitee.com/losschen/yolov5-pytorch/blob/main/README.md?

一、制作數(shù)據(jù)集文件

根目錄下創(chuàng)建數(shù)據(jù)集文件夾，類別文檔

圖片和標(biāo)注扔到Annotations和JPEGimages里面

把MAIN文件夾從voc數(shù)據(jù)集里復(fù)制過來

打開voc_Annotations.py

改路徑

由于文件夾取名GangMao2022，year2007改2022

每一個(gè)%s前的VOC都改成GangMao

改

保存運(yùn)行，報(bào)錯(cuò)

改成正確的標(biāo)簽名字

成功，新的文件信息生成了。

二、訓(xùn)練yolo模型

參考：

運(yùn)行報(bào)錯(cuò)

?在pytorch環(huán)境里面下載一下，pip install tensorboard?

報(bào)錯(cuò)

FileNotFoundError: [Errno 2] No such file or directory:?'2007_train.txt'

注意路徑不是數(shù)據(jù)集里那個(gè)，而是主目錄下新生成的2022_train.txt

里面的樣子

優(yōu)化方法名字打錯(cuò)了

adam

ok，成功運(yùn)行，推薦577，但是先跑100世代試一試。

?

昨晚跑了7個(gè)小時(shí)的結(jié)果，使用測試集的效果還可以但是換一般的圖片識(shí)別率就不行了，并且首片的識(shí)別率還是太低。