mirror of
https://github.com/ultralytics/yolov5.git
synced 2025-06-03 14:49:29 +08:00
Edge TPU TF imports fix (#6542)
* Edge TPU TF imports fix Fix for https://github.com/ultralytics/yolov5/issues/6535#issuecomment-1030631526 * Update common.py
This commit is contained in:
parent
cba4303d32
commit
dc7e093071
@ -374,11 +374,11 @@ class DetectMultiBackend(nn.Module):
|
|||||||
graph_def.ParseFromString(open(w, 'rb').read())
|
graph_def.ParseFromString(open(w, 'rb').read())
|
||||||
frozen_func = wrap_frozen_graph(gd=graph_def, inputs="x:0", outputs="Identity:0")
|
frozen_func = wrap_frozen_graph(gd=graph_def, inputs="x:0", outputs="Identity:0")
|
||||||
elif tflite: # https://www.tensorflow.org/lite/guide/python#install_tensorflow_lite_for_python
|
elif tflite: # https://www.tensorflow.org/lite/guide/python#install_tensorflow_lite_for_python
|
||||||
try: # prefer tflite_runtime if installed
|
try: # https://coral.ai/docs/edgetpu/tflite-python/#update-existing-tf-lite-code-for-the-edge-tpu
|
||||||
from tflite_runtime.interpreter import Interpreter, load_delegate
|
from tflite_runtime.interpreter import Interpreter, load_delegate
|
||||||
except ImportError:
|
except ImportError:
|
||||||
import tensorflow.lite.experimental.load_delegate as load_delegate
|
import tensorflow as tf
|
||||||
import tensorflow.lite.Interpreter as Interpreter
|
Interpreter, load_delegate = tf.lite.Interpreter, tf.lite.experimental.load_delegate,
|
||||||
if 'edgetpu' in w.lower(): # Edge TPU https://coral.ai/software/#edgetpu-runtime
|
if 'edgetpu' in w.lower(): # Edge TPU https://coral.ai/software/#edgetpu-runtime
|
||||||
LOGGER.info(f'Loading {w} for TensorFlow Lite Edge TPU inference...')
|
LOGGER.info(f'Loading {w} for TensorFlow Lite Edge TPU inference...')
|
||||||
delegate = {'Linux': 'libedgetpu.so.1',
|
delegate = {'Linux': 'libedgetpu.so.1',
|
||||||
|
Loading…
x
Reference in New Issue
Block a user