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