diff --git a/ultralytics/engine/exporter.py b/ultralytics/engine/exporter.py index 82675be620..20eb87d2a3 100644 --- a/ultralytics/engine/exporter.py +++ b/ultralytics/engine/exporter.py @@ -75,7 +75,6 @@ from ultralytics.utils import ( ARM64, DEFAULT_CFG, IS_JETSON, - IS_RASPBERRYPI, LINUX, LOGGER, MACOS, @@ -1012,7 +1011,7 @@ class Exporter: """Add metadata to *.tflite models per https://www.tensorflow.org/lite/models/convert/metadata.""" import flatbuffers - if MACOS or IS_RASPBERRYPI: # TODO: should this be 'if ARM64:'? + if ARM64: from tflite_support import metadata # noqa from tflite_support import metadata_schema_py_generated as schema # noqa else: