move out stuff from __init__

mikel-brostrom · Jul 8, 2024 · 2f297f3 · 2f297f3
1 parent d4fa0e9
commit 2f297f3
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/boxmot/appearance/reid_export.py b/boxmot/appearance/reid_export.py
@@ -11,9 +11,10 @@
 from torch.utils.mobile_optimizer import optimize_for_mobile
 
 from boxmot.appearance import export_formats
-from boxmot.appearance.backbones import build_model, get_nr_classes
 from boxmot.appearance.reid_model_factory import (get_model_name,
-                                                  load_pretrained_weights)
+                                                  load_pretrained_weights,build_model,
+                                                  get_nr_classes
+                                                 )
 from boxmot.utils import WEIGHTS
 from boxmot.utils import logger as LOGGER
 from boxmot.utils.checks import TestRequirements