diff --git a/plugins/tensorflow-lite/.vscode/settings.json b/plugins/tensorflow-lite/.vscode/settings.json index 1da3eb5ac7..e4506078e6 100644 --- a/plugins/tensorflow-lite/.vscode/settings.json +++ b/plugins/tensorflow-lite/.vscode/settings.json @@ -8,9 +8,13 @@ // "scrypted.debugHost": "192.168.2.119", // "scrypted.serverRoot": "/home/pi/.scrypted", + // lxc installation + "scrypted.debugHost": "scrypted-server", + "scrypted.serverRoot": "/root/.scrypted", + // local checkout - "scrypted.debugHost": "127.0.0.1", - "scrypted.serverRoot": "/Users/koush/.scrypted", + // "scrypted.debugHost": "127.0.0.1", + // "scrypted.serverRoot": "/Users/koush/.scrypted", // "scrypted.debugHost": "koushik-windows", // "scrypted.serverRoot": "C:\\Users\\koush\\.scrypted", diff --git a/plugins/tensorflow-lite/package-lock.json b/plugins/tensorflow-lite/package-lock.json index 3a1d8f336b..9afd4c84d7 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.1.47", + "version": "0.1.48", "lockfileVersion": 2, "requires": true, "packages": { "": { "name": "@scrypted/tensorflow-lite", - "version": "0.1.47", + "version": "0.1.48", "devDependencies": { "@scrypted/sdk": "file:../../sdk" } diff --git a/plugins/tensorflow-lite/package.json b/plugins/tensorflow-lite/package.json index 8b4c0014d3..0e2b29840f 100644 --- a/plugins/tensorflow-lite/package.json +++ b/plugins/tensorflow-lite/package.json @@ -53,5 +53,5 @@ "devDependencies": { "@scrypted/sdk": "file:../../sdk" }, - "version": "0.1.47" + "version": "0.1.48" } diff --git a/plugins/tensorflow-lite/src/tflite/__init__.py b/plugins/tensorflow-lite/src/tflite/__init__.py index bb9ce223a4..66b084c479 100644 --- a/plugins/tensorflow-lite/src/tflite/__init__.py +++ b/plugins/tensorflow-lite/src/tflite/__init__.py @@ -70,7 +70,13 @@ def configureModel(): nonlocal model if defaultModel: - model = "efficientdet_lite0_320_ptq" + if edge_tpus and next((obj for obj in edge_tpus if obj['type'] == 'usb'), None): + model = "yolov8n_full_integer_quant_320" + # this model seems completely wacky with lots of false positives. + # might be broken? + # model = "ssdlite_mobiledet_coco_qat_postprocess" + else: + model = "efficientdet_lite0_320_ptq" self.yolo = "yolo" in model self.yolov8 = "yolov8" in model