diff --git a/cv/detection/yolov5/pytorch/README.md b/cv/detection/yolov5/pytorch/README.md index c502c0393ad31cd607724f4a4cbce7f178131219..9f230bf8f835e3a09bb868d14dfab050a18ae0ad 100644 --- a/cv/detection/yolov5/pytorch/README.md +++ b/cv/detection/yolov5/pytorch/README.md @@ -47,22 +47,22 @@ Train the yolov5 model as follows, the train log is saved in ./runs/train/exp ### On single GPU $ cd yolov5 - $ python3 train.py --data ./data/coco.yaml --batch-size 32 --cfg ./models/yolov5m.yaml + $ python3 train.py --data ./data/coco.yaml --batch-size 32 --cfg ./models/yolov5s.yaml ### On single GPU (AMP) - $ python3 train.py --data ./data/coco.yaml --batch-size 32 --cfg ./models/yolov5m.yaml --amp + $ python3 train.py --data ./data/coco.yaml --batch-size 32 --cfg ./models/yolov5s.yaml --amp ### Multiple GPUs on one machine $ # eight cards - $ python3 -m torch.distributed.launch --nproc_per_node 8 train.py --data ./data/coco.yaml --batch-size 256 --cfg ./models/yolov5m.yaml --device 0,1,2,3,4,5,6,7 # bash run_dist_training.sh + $ python3 -m torch.distributed.launch --nproc_per_node 8 train.py --data ./data/coco.yaml --batch-size 256 --cfg ./models/yolov5s.yaml --device 0,1,2,3,4,5,6,7 ### Multiple GPUs on one machine (AMP) $ # eight cards - $ python3 -m torch.distributed.launch --nproc_per_node 8 train.py --data ./data/coco.yaml --batch-size 256 --cfg ./models/yolov5m.yaml --device 0,1,2,3,4,5,6,7 --amp + $ python3 -m torch.distributed.launch --nproc_per_node 8 train.py --data ./data/coco.yaml --batch-size 256 --cfg ./models/yolov5s.yaml --device 0,1,2,3,4,5,6,7 --amp ## Test the detector @@ -70,8 +70,8 @@ Train the yolov5 model as follows, the train log is saved in ./runs/train/exp Test the yolov5 model as follows, the result is saved in ./runs/detect: $ cd yolov5 - $ python3 detect.py --source ./data/images/bus.jpg --weight yolov5m.pt --img 640 - $ python3 detect.py --source ./data/images/zidane.jpg --weight yolov5m.pt --img 640 + $ python3 detect.py --source ./data/images/bus.jpg --weight yolov5s.pt --img 640 + $ python3 detect.py --source ./data/images/zidane.jpg --weight yolov5s.pt --img 640 ## Results on BI-V100