UltralyticsAssistant 5 months ago
parent 24f2cfe61a
commit 629e2c59b1
  1. 2
      ultralytics/engine/exporter.py
  2. 2
      ultralytics/nn/autobackend.py

@ -1041,7 +1041,7 @@ class Exporter:
@try_export
def export_mct(self, prefix=colorstr("Sony MCT:")):
check_requirements(["mct-nightly","sony-custom-layers[torch]"])
check_requirements(["mct-nightly", "sony-custom-layers[torch]"])
import model_compression_toolkit as mct
import onnx
from model_compression_toolkit.core import BitWidthConfig

@ -178,7 +178,7 @@ class AutoBackend(nn.Module):
providers = ["CUDAExecutionProvider", "CPUExecutionProvider"] if cuda else ["CPUExecutionProvider"]
if mct:
check_requirements(["mct-nightly","sony-custom-layers[torch]"])
check_requirements(["mct-nightly", "sony-custom-layers[torch]"])
LOGGER.info(f"Loading {w} for ONNX MCT quantization inference...")
import mct_quantizers as mctq
from sony_custom_layers.pytorch.object_detection import nms_ort # noqa

Loading…
Cancel
Save