模型转换onnx2engine

pytorch2onnx

python demo_pytorch2onnx.py <weight_file> <image_path> <batch_size> <n_classes> <IN_IMAGE_H> <IN_IMAGE_W>

example

python demo_pytorch2onnx.py yolov4.pth dog.jpg 8 80 416 416

onnx2engine

trtexec --onnx=<onnx_file> --explicitBatch --saveEngine=<tensorRT_engine_file> --workspace=<size_in_megabytes> --fp16

example

./trtexec --explicitBatch --onnx=./resnet50.onnx --saveEngine=resnet50.engine