Auto-format by https://ultralytics.com/actions

4 months ago · 9b066bacb2
parent 85b236c651
commit 9b066bacb2
1 changed files with 3 additions and 3 deletions
--- a/ultralytics/engine/exporter.py
+++ b/ultralytics/engine/exporter.py
@ -1111,8 +1111,8 @@ class Exporter:
            raise ValueError("IMX500 export is only supported for YOLOv8 detection models")
        check_requirements(("model-compression-toolkit==2.1.1", "sony-custom-layers[torch]"))
        import model_compression_toolkit as mct
-        from sony_custom_layers.pytorch.object_detection.nms import multiclass_nms
        import onnx
+        from sony_custom_layers.pytorch.object_detection.nms import multiclass_nms

        def representative_dataset_gen(dataloader=self.get_int8_calibration_dataloader(prefix)):
            for batch in dataloader: