Skip MNN export for Raspberry Pi and NVIDIA Jetson (#17292)

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
pull/17296/head^2
Lakshantha Dissanayake 3 weeks ago committed by GitHub
parent 1e70710f3e
commit daaac35fff
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
  1. 1
      tests/test_exports.py
  2. 3
      ultralytics/engine/exporter.py

@ -193,6 +193,7 @@ def test_export_paddle():
@pytest.mark.slow
@pytest.mark.skipif(IS_RASPBERRYPI, reason="MNN not supported on Raspberry Pi")
def test_export_mnn():
"""Test YOLO exports to MNN format (WARNING: MNN test must precede NCNN test or CI error on Windows)."""
file = YOLO(MODEL).export(format="mnn", imgsz=32)

@ -77,6 +77,7 @@ from ultralytics.utils import (
ARM64,
DEFAULT_CFG,
IS_JETSON,
IS_RASPBERRYPI,
LINUX,
LOGGER,
MACOS,
@ -244,6 +245,8 @@ class Exporter:
"WARNING ⚠ INT8 export requires a missing 'data' arg for calibration. "
f"Using default 'data={self.args.data}'."
)
if mnn and (IS_RASPBERRYPI or IS_JETSON):
raise SystemError("MNN export not supported on Raspberry Pi and NVIDIA Jetson")
# Input
im = torch.zeros(self.args.batch, 3, *self.imgsz).to(self.device)
file = Path(

Loading…
Cancel
Save