From c28e60d875bc0da676c13a1e156fd70e17f24bf2 Mon Sep 17 00:00:00 2001 From: Koushik Dutta Date: Mon, 29 Jan 2024 21:02:00 -0800 Subject: [PATCH] tensorflow-lite: switch default model to efficientdet_lite0_320_ptq --- plugins/tensorflow-lite/package-lock.json | 4 ++-- plugins/tensorflow-lite/package.json | 2 +- plugins/tensorflow-lite/src/tflite/__init__.py | 2 +- 3 files changed, 4 insertions(+), 4 deletions(-) diff --git a/plugins/tensorflow-lite/package-lock.json b/plugins/tensorflow-lite/package-lock.json index f52f13b333..588ebc9efe 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.45", + "version": "0.1.46", "lockfileVersion": 2, "requires": true, "packages": { "": { "name": "@scrypted/tensorflow-lite", - "version": "0.1.45", + "version": "0.1.46", "devDependencies": { "@scrypted/sdk": "file:../../sdk" } diff --git a/plugins/tensorflow-lite/package.json b/plugins/tensorflow-lite/package.json index 320ee50b3c..0c6e9e3d54 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.45" + "version": "0.1.46" } diff --git a/plugins/tensorflow-lite/src/tflite/__init__.py b/plugins/tensorflow-lite/src/tflite/__init__.py index bf2ee32933..bb9ce223a4 100644 --- a/plugins/tensorflow-lite/src/tflite/__init__.py +++ b/plugins/tensorflow-lite/src/tflite/__init__.py @@ -70,7 +70,7 @@ def configureModel(): nonlocal model if defaultModel: - model = "yolov8n_full_integer_quant_320" + model = "efficientdet_lite0_320_ptq" self.yolo = "yolo" in model self.yolov8 = "yolov8" in model