yolov5-v3.0尝鲜

README

20200817, yolov5发布了3.0，精度来到了49.2，加上多尺度tta后 50.8

安装

感觉这个issue, v3.0的release指向的分支有问题，所以保险起见

git clone [email protected]:ultralytics/yolov5.git
git reset --hard 916d4aa

下载weights

bash weights/download_weights.sh

不要用github里面的release来下载，太慢，用这个比较快

文件布局

yolov5的文件布局我不习惯，比如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对应上了

test

python test.py --data coco/data.yaml --weights weights/yolov5x.pt  --img 640 --conf 0.001 --save-json

trouble shooting

进度条给的map低了点

是的， 根据issue yolov5 test.py里面的评估有点问题，加上--save-json调用pycocotools来评估就可以了