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

Fix TFLite export for NVIDIA Jetson (#10404)

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-01 06:56:30 -07:00 • committed by GitHub
parent 15060e13cf
commit bbfca94467
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 2 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

3
ultralytics/engine/exporter.py
View file

@ -73,6 +73,7 @@ from ultralytics.nn.tasks import DetectionModel, SegmentationModel, WorldModel
from ultralytics.utils import ( from ultralytics.utils import (
ARM64, ARM64,
DEFAULT_CFG, DEFAULT_CFG,
IS_JETSON,
LINUX, LINUX,
LOGGER, LOGGER,
MACOS, MACOS,
@ -749,7 +750,7 @@ class Exporter:
"sng4onnx>=1.0.1", "sng4onnx>=1.0.1",
"onnxsim>=0.4.33", "onnxsim>=0.4.33",
"onnx_graphsurgeon>=0.3.26", "onnx_graphsurgeon>=0.3.26",
"tflite_support", "tflite_support<=0.4.3" if IS_JETSON else "tflite_support", # fix ImportError 'GLIBCXX_3.4.29'
"flatbuffers>=23.5.26,<100", # update old 'flatbuffers' included inside tensorflow package "flatbuffers>=23.5.26,<100", # update old 'flatbuffers' included inside tensorflow package
"onnxruntime-gpu" if cuda else "onnxruntime", "onnxruntime-gpu" if cuda else "onnxruntime",
), ),