README
20200817, yolov5發(fā)布了3.0擒悬,精度來到了49.2,加上多尺度tta后 50.8
安裝
感覺這個issue, v3.0的release指向的分支有問題,所以保險起見
git clone git@github.com:ultralytics/yolov5.git
git reset --hard 916d4aa
下載weights
bash weights/download_weights.sh
不要用github里面的release來下載猫十,太慢蚓再,用這個比較快
文件布局
yolov5的文件布局我不習(xí)慣,比如data里面放yaml, coco路徑和yolov5同級一姿,有點別扭七咧,我改了一下,在yolov5下新建coco文件夾叮叹,如下艾栋,
coco
|-- data.yaml
`-- model.yaml
model.yaml
就是yolov5x.yaml
,data.yaml
改了一下路徑
然后進入
data/scripts
蛉顽,運行
bash get_coco.sh
會得到data/coco
這個文件夾蝗砾,和data.yaml
對應(yīng)上了
test
python test.py --data coco/data.yaml --weights weights/yolov5x.pt --img 640 --conf 0.001 --save-json
可以復(fù)現(xiàn)49.2的map,挺好
水下檢測實戰(zhàn)
python train.py --cfg underwater/model.yaml --data underwater/data.yaml --epochs 100 --weights weights/yolov5x.pt --batch-size 8
5260張 train 2*2080ti
, 顯存2*9300mb
,6min/epoch悼粮,設(shè)置的100epoch差不多是10h闲勺,有點久
map 0.5, map50 0.847,但是同數(shù)據(jù)的universe 0.523 0.858, cas 64 4d 0.536 0.867扣猫,可能是尺度小了點菜循,
在超算上試一下
python train.py --cfg underwater/model.yaml --data underwater/data.yaml --epochs 100 --weights weights/yolov5x.pt --batch-size 8 --cache-images
2V100(16G) 加上 cache
也是6min/epoch,此時是bs8 * 2申尤,換成bs 16 看看速度會不會加快癌幕,換成了bs 16 速度是4min/epoch,還可以昧穿,
bs4 1024 2gpu, 10min/epoch
換成4V100 bs 6勺远,6不是4的倍數(shù),報了錯粤咪,改成了4*bs4 谚中,還是10min/epoch,總共跑了18h寥枝,map 0.503宪塔,搞了一點點,map50最高是0.845囊拜,欸某筐,垃圾
trouble shooting
進度條給的map低了點
是的, 根據(jù)issue yolov5 test.py里面的評估有點問題冠跷,加上--save-json
調(diào)用pycocotools來評估就可以了