YOLOv5-Lite 测试过程
1.代码:https://github.com/ppogg/YOLOv5-Lite
参考网址:各类图像数据集下载地址 - DataSet Download
需要下载:
2017 Train images [118K/18GB]:http://images.cocodataset.org/zips/train2017.zip
2017 Val images [5K/1GB]:http://images.cocodataset.org/zips/val2017.zip
2017 Test images [41K/6GB]:http://images.cocodataset.org/zips/test2017.zip
2017 Train/Val annotations [241MB]:
http://images.cocodataset.org/annotations/annotations_trainval2017.zip
3.下载模型
模型找了很久,一直在找weight文件夹没找到,在下图所示位置下载模型
v5lite-s.pt:
https://drive.google.com/file/d/1zkKrJsfUx9WVKuFpPQgDemzf2geAHEx-/view
v5lite-c.pt:
https://drive.google.com/file/d/1H-GrrxKV0pN3Y9pH1-BNNO2_dpBjhd2R/view
v5lite-g.pt:
https://drive.google.com/file/d/1y6R1W9NtqvJgVeUp5VMmiFGn3sOFN2C1/view
下载完模型后,在项目中新建weight文件夹,把模型放入
4.配置环境
cd YOLOv5-Lite #放项目的路径下
pip install -r requirements.txt
5.测试需要数据,先放一张图片进行测试
将ncnn文件夹下images文件夹复制倒data文件夹下,images文件夹下只有一张图片用来测试
复制命令:
cp dir1/a.doc dir2 表示将dir1下的a.doc文件复制到dir2目录下
cp -r dir1 dir2 表示将dir1及其dir1下所包含的文件复制到dir2下
cp -r dir1/. dir2 表示将dir1下的文件复制到dir2,不包括dir1目录
6.测试
python detect.py --source data/images/ --weights weight/v5lite-s.pt
测试结果会生成一个run文件夹,结果在run/detect/exp
测试结果如下: