Edge TPU compiler comment (#6196)
* Edge TPU compiler comment * 7 to 8 fix
This commit is contained in:
1 parent
9e9219fe17
commit
5402753a53
2 files changed
+5
-5
No files matched your search
+2
-2
@@ -376,8 +376,8 @@ class DetectMultiBackend(nn.Module):
|
||||
elif tflite: # https://www.tensorflow.org/lite/guide/python#install_tensorflow_lite_for_python
|
||||
if 'edgetpu' in w.lower(): # Edge TPU
|
||||
LOGGER.info(f'Loading {w} for TensorFlow Lite Edge TPU inference...')
|
||||
import tflite_runtime.interpreter as tfli
|
||||
delegate = {'Linux': 'libedgetpu.so.1', # install https://coral.ai/software/#edgetpu-runtime
|
||||
import tflite_runtime.interpreter as tfli # install https://coral.ai/software/#edgetpu-runtime
|
||||
delegate = {'Linux': 'libedgetpu.so.1',
|
||||
'Darwin': 'libedgetpu.1.dylib',
|
||||
'Windows': 'edgetpu.dll'}[platform.system()]
|
||||
interpreter = tfli.Interpreter(model_path=w, experimental_delegates=[tfli.load_delegate(delegate)])
|
||||
|
||||
Reference in new issue
Block a user