Translate

Some commands for YOLOv5

1. train yolov5 using multi-gpu:

python -m torch.distributed.run --nproc_per_node 8 train.py --batch 64 --epochs 400 
--data ./data/fall.yaml --weights yolov5m.pt --device 0,1,2,3,4,5,6,7


2. yolov5 convert pytorch to onnx:

python export.py --weights yolov5m.pt --include onnx --dynamic

3. yolov5 convert onnx to trt:

/usr/src/tensorrt/bin/trtexec --onnx=./fall_det/1/0620.onnx --saveEngine=./fall_det/1/model.plan --workspace=8192 --explicitBatch --fp16 --verbose --dumpOutput --minShapes='images':1x3x640x640 --optShapes='images':10x3x640x640  --maxShapes='images':10x3x640x640





No comments:

Post a Comment