diff --git a/plugins/tensorflow-lite/package-lock.json b/plugins/tensorflow-lite/package-lock.json index d9792477d..f649390a5 100644 --- a/plugins/tensorflow-lite/package-lock.json +++ b/plugins/tensorflow-lite/package-lock.json @@ -1,12 +1,12 @@ { "name": "@scrypted/tensorflow-lite", - "version": "0.0.39", + "version": "0.0.40", "lockfileVersion": 2, "requires": true, "packages": { "": { "name": "@scrypted/tensorflow-lite", - "version": "0.0.39", + "version": "0.0.40", "hasInstallScript": true, "devDependencies": { "@scrypted/sdk": "file:../../sdk" diff --git a/plugins/tensorflow-lite/package.json b/plugins/tensorflow-lite/package.json index 27bec6801..65657a236 100644 --- a/plugins/tensorflow-lite/package.json +++ b/plugins/tensorflow-lite/package.json @@ -42,5 +42,5 @@ "devDependencies": { "@scrypted/sdk": "file:../../sdk" }, - "version": "0.0.39" + "version": "0.0.40" } diff --git a/plugins/tensorflow-lite/src/tflite/__init__.py b/plugins/tensorflow-lite/src/tflite/__init__.py index aa506adab..03e9fbea4 100644 --- a/plugins/tensorflow-lite/src/tflite/__init__.py +++ b/plugins/tensorflow-lite/src/tflite/__init__.py @@ -13,8 +13,9 @@ try: from pycoral.utils.edgetpu import list_edge_tpus from pycoral.utils.edgetpu import make_interpreter loaded_py_coral = True + print('coral edge tpu library loaded successfully') except Exception as e: - print('coral utils load failed', e) + print('coral edge tpu library load failed', e) pass import tflite_runtime.interpreter as tflite import re @@ -62,7 +63,8 @@ class TensorFlowLitePlugin(DetectPlugin): model = scrypted_sdk.zip.open( 'fs/mobilenet_ssd_v2_coco_quant_postprocess_edgetpu.tflite').read() self.interpreter = make_interpreter(model) - except: + except Exception as e: + print('unable to use Coral Edge TPU', e) self.edge_tpu_found = 'Edge TPU not found' model = scrypted_sdk.zip.open( 'fs/mobilenet_ssd_v2_coco_quant_postprocess.tflite').read()