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