瀏覽代碼

Update CPU model to Mobilenet v2

Inference speed went from ~470ms to ~530ms, however average confidence went from ~75% to ~90%+
mattheys 4 年之前
父節點
當前提交
b345571a63
共有 1 個文件被更改,包括 2 次插入4 次删除
  1. 2 4
      Dockerfile

+ 2 - 4
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/