Chiebot-Mirror
/
ultralytics
mirror of https://github.com/ultralytics/ultralytics.git
9
0
Fork 1
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source
pull/17463/head^2
UltralyticsAssistant 1 month ago
parent 478d126c07
commit 5abb248dcd
1 changed files with 1 additions and 1 deletions
Whitespace
Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL
Unified View
Diff Options
Show Stats Download Patch File Download Diff File
  1. 2
      ultralytics/engine/exporter.py

2
ultralytics/engine/exporter.py
Unescape View File

@ -1107,8 +1107,8 @@ class Exporter:
if "C2f" not in self.model.__str__(): if "C2f" not in self.model.__str__():
raise ValueError("IMX500 export is only supported for YOLOv8 detection models") raise ValueError("IMX500 export is only supported for YOLOv8 detection models")
check_requirements(("model-compression-toolkit==2.1.1", "sony-custom-layers[torch]")) 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 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)): def representative_dataset_gen(dataloader=self.get_int8_calibration_dataloader(prefix)):
for batch in dataloader: for batch in dataloader:

Write Preview
Loading…
Cancel
Save