removed nms=False support

test-quan
Francesco Mattioli 4 weeks ago
parent 4182ddb63d
commit 354ea825e8
  1. 7
      ultralytics/engine/exporter.py

@ -1106,7 +1106,8 @@ class Exporter:
raise ValueError("MCT export is not supported for end2end models.")
if "C2f" not in self.model.__str__():
raise ValueError("MCT export is only supported for YOLOv8 detection models")
check_requirements("model-compression-toolkit==2.1.1")
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
def representative_dataset_gen(dataloader=self.get_int8_calibration_dataloader(prefix)):
@ -1145,10 +1146,6 @@ class Exporter:
)[0]
)
if self.args.nms:
check_requirements("sony-custom-layers[torch]")
from sony_custom_layers.pytorch.object_detection.nms import multiclass_nms
class NMSWrapper(torch.nn.Module):
def __init__(
self,

Loading…
Cancel
Save