yolov5 复现
github:https://github.com/ultralytics/yolov5
安装依赖
pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple
数据集 同yolov7
修改类别
models/yolov5s.yaml
train.py
命令
python train.py --data data/mycoco.yaml --cfg yolov5s.yaml --weights yolov5s.pt --epochs 100 --batch-size 1