Improve `tflite` exports for Tests CI (#13094)

pull/13099/head
Lakshantha Dissanayake 8 months ago committed by GitHub
parent 027d4a6304
commit f3bf77b416
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
  1. 3
      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:

Loading…
Cancel
Save