pytorch2onnx
python demo_pytorch2onnx.py
example
python demo_pytorch2onnx.py yolov4.pth dog.jpg 8 80 416 416
onnx2engine
trtexec --onnx=--explicitBatch --saveEngine= --workspace= --fp16
example
./trtexec --explicitBatch --onnx=./resnet50.onnx --saveEngine=resnet50.engine
