1.代码:https://github.com/ppogg/YOLOv5-Lite
2.模型:点v5-Lite.pt下载
3.使用coco128数据集进行训练
因为coco2017太大,先用比较小的coco128数据集进行训练看看
下载完之后与代码文件夹放在同一个目录下
环境:
source activate yolov5lite1
测试:
python detect.py --source data/images/ --weights weight/v5lite-s.pt
训练:
python train.py --data data/coco128.yaml --cfg models/v5Lite-s.yaml --weights weight/v5lite-s.pt --batch-size 8