|
|
@ -75,7 +75,6 @@ from ultralytics.utils import ( |
|
|
|
ARM64, |
|
|
|
ARM64, |
|
|
|
DEFAULT_CFG, |
|
|
|
DEFAULT_CFG, |
|
|
|
IS_JETSON, |
|
|
|
IS_JETSON, |
|
|
|
IS_RASPBERRYPI, |
|
|
|
|
|
|
|
LINUX, |
|
|
|
LINUX, |
|
|
|
LOGGER, |
|
|
|
LOGGER, |
|
|
|
MACOS, |
|
|
|
MACOS, |
|
|
@ -1012,7 +1011,7 @@ class Exporter: |
|
|
|
"""Add metadata to *.tflite models per https://www.tensorflow.org/lite/models/convert/metadata.""" |
|
|
|
"""Add metadata to *.tflite models per https://www.tensorflow.org/lite/models/convert/metadata.""" |
|
|
|
import flatbuffers |
|
|
|
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 # noqa |
|
|
|
from tflite_support import metadata_schema_py_generated as schema # noqa |
|
|
|
from tflite_support import metadata_schema_py_generated as schema # noqa |
|
|
|
else: |
|
|
|
else: |
|
|
|