旅行青蛙
Published on 2025-01-14 / 8 Visits
0
0

Yolov 模型训练

Yolo5 模型训练

第一步下载

https://github.com/ultralytics/yolov5

第二步安装

cd yolov5

pip install -r requirements.txt

第三步测试

python detect.py --source data/images/bus.jpg --weights pretrained/yolov5s.pt

第四步检查

 cd runs/detect/exp/

看看有没有一个bus.jpg,如果有,恭喜你,成功了

第五步准备数据集

第六步训练数据

python train.py --data tyc.yaml --cfg yolov5n.yaml --weights pretrained/yolov5n.pt --epoch 100 --batch-size 4 --device cpu

第七步 导出onnx模型

python3 export.py --weights /opt/yolov5/runs/train/exp12/weights/last.pt --include onnx --dynamic

java selenium判断页面加载完成

https://blog.51cto.com/u_16099277/6362273

YOLOv7训练

https://blog.csdn.net/qing2019/article/details/127054577

https://blog.csdn.net/weixin_43687366/article/details/126166982


Comment