open-mmlab / mmdetection

OpenMMLab Detection Toolbox and Benchmark
https://mmdetection.readthedocs.io
Apache License 2.0
28.47k stars 9.28k forks source link

Functionality to combine ONNX models with non-overlapping class lists #11781

Open GeorgePearse opened 2 weeks ago

GeorgePearse commented 2 weeks ago

Describe the feature

Motivation I very often train models on different subsets of our dataset, it'd be great if I could then recombine these for running inference with Triton. I freeze a few stages of the backbone (and would happily freeze more to enable this).

The different models have non-overlapping class lists