Skip to content

ReID multi framework model export

Jump to bottom
Mike edited this page Jan 5, 2023 · 7 revisions

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

python3 trackers/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.

Clone this wiki locally