-
-
Notifications
You must be signed in to change notification settings - Fork 1.8k
ReID multi framework model export
Mike edited this page Nov 17, 2023
·
7 revisions
Exporting your ReID models to TorchScript, ONNX, OpenVINO and TensorRT is as simple as:
python3 boxmot/appearance/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.