1.准备数据集
2.训练
python .\train.py --workers 1 --device 0 --batch-size 8 --epochs 100 --img 640 640 --data .\data\elephdev.yaml --hyp .\data\hyp.scratch.custom.yaml --cfg
.\cfg\training\elephdev.yaml --name yolov7-elephdev --weights .\yolov7-tiny.pt
3.查看训练结果统计
tensorboard --logdir runs/train
浏览器访问: http://localhost:6006/
4.测试
python detect.py --weights .\elephdev.pt --conf 0.25 --img-size 640 --source .\inference\test\elephdev.jpg
发表评论 取消回复