From 5abb248dcd75b4752a5d7e1fe6099631a05c0775 Mon Sep 17 00:00:00 2001 From: UltralyticsAssistant Date: Wed, 6 Nov 2024 10:51:54 +0000 Subject: [PATCH] Auto-format by https://ultralytics.com/actions --- ultralytics/engine/exporter.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ultralytics/engine/exporter.py b/ultralytics/engine/exporter.py index 15b52ab2e..643cbdb35 100644 --- a/ultralytics/engine/exporter.py +++ b/ultralytics/engine/exporter.py @@ -1107,8 +1107,8 @@ class Exporter: if "C2f" not in self.model.__str__(): raise ValueError("IMX500 export is only supported for YOLOv8 detection models") check_requirements(("model-compression-toolkit==2.1.1", "sony-custom-layers[torch]")) - from sony_custom_layers.pytorch.object_detection.nms import multiclass_nms import model_compression_toolkit as mct + 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: