yolov5训练bdd100k
时间: 2023-09-16 15:12:23 浏览: 117
您可以使用以下指令来训练yolov7模型使用bdd100k数据集:
python train.py --workers 0 --device 0 --batch-size 4 --data data/bdd100k.yaml --img 640 640 --cfg cfg/training/yolov7.yaml --weights '' --name yolov7 --hyp data/hyp.scratch.p5.yaml
如果您希望进行双卡训练,可以使用以下命令:
python -m torch.distributed.launch --nproc_per_node 2 --master_port 25000 train.py --workers 1 --device 0,1 --sync-bn --batch-size 2 --data data/bdd100k.yaml --img 640 640 --cfg cfg/training/yolov7_bdd100k.yaml --weights '' --name bdd100k --hyp data/hyp.bdd100k.yaml --epochs 12
在训练完成后,您可以使用以下指令进行测试,其中需要指定权重文件的位置:
python test.py --data data/bdd100k.yaml --img 640 --batch 1 --conf 0.001 --iou 0.65 --device 0 --weights runs/train/bdd100k/weights/best.pt --name yolov7_bdd100k <span class="em">1</span><span class="em">2</span><span class="em">3</span>