HongChuang/ultralytics-ascend
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 3

Fix Failing RPi CI (#13088)

Browse source 
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
This commit is contained in:
Lakshantha Dissanayake 2024-05-23 22:47:12 -07:00 • committed by GitHub
parent 11623eeb00
commit 027d4a6304
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 4 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

2
.github/workflows/ci.yaml vendored
View file

@ -253,7 +253,7 @@ jobs:
- name: Install requirements - name: Install requirements
run: | run: |
python -m pip install --upgrade pip wheel python -m pip install --upgrade pip wheel
pip install -e ".[export]" pytest pip install --no-cache-dir -e ".[export]" pytest mlflow pycocotools "ray[tune]"
- name: Check environment - name: Check environment
run: | run: |
yolo checks yolo checks

4
ultralytics/engine/exporter.py
View file

@ -75,6 +75,7 @@ from ultralytics.utils import (
ARM64, ARM64,
DEFAULT_CFG, DEFAULT_CFG,
IS_JETSON, IS_JETSON,
IS_RASPBERRYPI,
LINUX, LINUX,
LOGGER, LOGGER,
MACOS, MACOS,
@ -1011,10 +1012,11 @@ 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: # TFLite Support bug https://github.com/tensorflow/tflite-support/issues/954#issuecomment-2108570845 if MACOS or IS_RASPBERRYPI: # TODO: should this be '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:
# TFLite Support bug https://github.com/tensorflow/tflite-support/issues/954#issuecomment-2108570845
from tensorflow_lite_support.metadata import metadata_schema_py_generated as schema # noqa from tensorflow_lite_support.metadata import metadata_schema_py_generated as schema # noqa
from tensorflow_lite_support.metadata.python import metadata # noqa from tensorflow_lite_support.metadata.python import metadata # noqa