当前位置：网站首页>模型转换onnx2engine

模型转换onnx2engine

2022-07-03 03:06:00 【AphilGuo】

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

版权声明
本文为[AphilGuo]所创，转载请带上原文链接，感谢
https://blog.csdn.net/qq_44798484/article/details/125496148