From b345571a6387c5b6b1bc34504792539710691805 Mon Sep 17 00:00:00 2001 From: mattheys Date: Thu, 2 Jul 2020 14:43:33 +0100 Subject: [PATCH] Update CPU model to Mobilenet v2 Inference speed went from ~470ms to ~530ms, however average confidence went from ~75% to ~90%+ --- Dockerfile | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/Dockerfile b/Dockerfile index dc92261c3..71ffc2b6f 100755 --- a/Dockerfile +++ b/Dockerfile @@ -46,10 +46,8 @@ RUN apt -qq update && apt -qq install --no-install-recommends -y \ # get model and labels RUN wget -q https://github.com/google-coral/edgetpu/raw/master/test_data/mobilenet_ssd_v2_coco_quant_postprocess_edgetpu.tflite -O /edgetpu_model.tflite --trust-server-names RUN wget -q https://dl.google.com/coral/canned_models/coco_labels.txt -O /labelmap.txt --trust-server-names -RUN wget -q https://storage.googleapis.com/download.tensorflow.org/models/tflite/coco_ssd_mobilenet_v1_1.0_quant_2018_06_29.zip -O /cpu_model.zip && \ - unzip /cpu_model.zip detect.tflite -d / && \ - mv /detect.tflite /cpu_model.tflite && \ - rm /cpu_model.zip +RUN wget -q https://github.com/google-coral/edgetpu/raw/master/test_data/mobilenet_ssd_v2_coco_quant_postprocess.tflite -O /cpu_model.tflite + WORKDIR /opt/frigate/ ADD frigate frigate/