From cb476d2999692bf64b033e0194afa18c3ec4eec1 Mon Sep 17 00:00:00 2001 From: Lakshantha Dissanayake Date: Tue, 19 Nov 2024 00:41:15 +0800 Subject: [PATCH] Enable MNN on RPi and Jetson (#17583) Co-authored-by: UltralyticsAssistant --- ultralytics/engine/exporter.py | 3 --- ultralytics/utils/benchmarks.py | 2 -- 2 files changed, 5 deletions(-) diff --git a/ultralytics/engine/exporter.py b/ultralytics/engine/exporter.py index c618e794..211282f7 100644 --- a/ultralytics/engine/exporter.py +++ b/ultralytics/engine/exporter.py @@ -79,7 +79,6 @@ from ultralytics.utils import ( ARM64, DEFAULT_CFG, IS_JETSON, - IS_RASPBERRYPI, LINUX, LOGGER, MACOS, @@ -265,8 +264,6 @@ 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) diff --git a/ultralytics/utils/benchmarks.py b/ultralytics/utils/benchmarks.py index 24e8ea9a..e65d1288 100644 --- a/ultralytics/utils/benchmarks.py +++ b/ultralytics/utils/benchmarks.py @@ -114,8 +114,6 @@ def benchmark( assert LINUX or MACOS, "Windows Paddle exports not supported yet" if i == 12: # MNN assert not isinstance(model, YOLOWorld), "YOLOWorldv2 MNN exports not supported yet" - assert not IS_RASPBERRYPI, "MNN export not supported on Raspberry Pi" - assert not IS_JETSON, "MNN export not supported on NVIDIA Jetson" if i == 13: # NCNN assert not isinstance(model, YOLOWorld), "YOLOWorldv2 NCNN exports not supported yet" if i == 14: # IMX