Skip to content

ReID multi framework model export

Mike edited this page Oct 28, 2022 · 7 revisions

Exporting your ReID models to TorchScript, ONNX, OpenVINO and TensorRT is as simple as:

python3 reid_export.py --include torchscript onnx openvino engine --device 0 --batch-size <max_num_expected_objects> --dynamic

This script takes in a PT model and converts it the selected frameworks. The exported models can take in an arbitrary number of inputs.