2020-01-16 17:14:58 -06:00
|
|
|
FROM ubuntu:18.04
|
2020-01-05 17:43:14 -06:00
|
|
|
LABEL maintainer "blakeb@blakeshome.com"
|
2019-06-02 07:29:50 -05:00
|
|
|
|
2020-01-05 17:43:14 -06:00
|
|
|
ENV DEBIAN_FRONTEND=noninteractive
|
2019-05-27 10:17:57 -05:00
|
|
|
# Install packages for apt repo
|
2020-01-05 17:43:14 -06:00
|
|
|
RUN apt -qq update && apt -qq install --no-install-recommends -y \
|
2020-01-28 07:24:02 -06:00
|
|
|
software-properties-common \
|
|
|
|
|
# apt-transport-https ca-certificates \
|
|
|
|
|
build-essential \
|
2020-05-01 06:59:24 -05:00
|
|
|
gnupg wget unzip tzdata \
|
2020-01-28 07:24:02 -06:00
|
|
|
# libcap-dev \
|
|
|
|
|
&& add-apt-repository ppa:deadsnakes/ppa -y \
|
|
|
|
|
&& apt -qq install --no-install-recommends -y \
|
|
|
|
|
python3.7 \
|
|
|
|
|
python3.7-dev \
|
|
|
|
|
python3-pip \
|
|
|
|
|
ffmpeg \
|
|
|
|
|
# VAAPI drivers for Intel hardware accel
|
|
|
|
|
libva-drm2 libva2 i965-va-driver vainfo \
|
2020-09-07 12:18:03 -05:00
|
|
|
&& python3.7 -m pip install -U pip \
|
2020-01-28 07:24:02 -06:00
|
|
|
&& python3.7 -m pip install -U wheel setuptools \
|
|
|
|
|
&& python3.7 -m pip install -U \
|
|
|
|
|
opencv-python-headless \
|
|
|
|
|
# python-prctl \
|
|
|
|
|
numpy \
|
|
|
|
|
imutils \
|
2020-01-31 11:50:42 -05:00
|
|
|
scipy \
|
2020-07-09 06:57:16 -05:00
|
|
|
psutil \
|
2020-01-31 11:50:42 -05:00
|
|
|
&& python3.7 -m pip install -U \
|
2020-02-15 21:07:54 -06:00
|
|
|
Flask \
|
|
|
|
|
paho-mqtt \
|
|
|
|
|
PyYAML \
|
|
|
|
|
matplotlib \
|
|
|
|
|
pyarrow \
|
2020-09-07 12:17:42 -05:00
|
|
|
click \
|
2020-01-05 17:43:14 -06:00
|
|
|
&& echo "deb https://packages.cloud.google.com/apt coral-edgetpu-stable main" > /etc/apt/sources.list.d/coral-edgetpu.list \
|
|
|
|
|
&& wget -q -O - https://packages.cloud.google.com/apt/doc/apt-key.gpg | apt-key add - \
|
|
|
|
|
&& apt -qq update \
|
|
|
|
|
&& echo "libedgetpu1-max libedgetpu/accepted-eula boolean true" | debconf-set-selections \
|
|
|
|
|
&& apt -qq install --no-install-recommends -y \
|
2020-01-28 07:24:02 -06:00
|
|
|
libedgetpu1-max \
|
|
|
|
|
## Tensorflow lite (python 3.7 only)
|
2020-03-02 06:11:19 -06:00
|
|
|
&& wget -q https://dl.google.com/coral/python/tflite_runtime-2.1.0.post1-cp37-cp37m-linux_x86_64.whl \
|
|
|
|
|
&& python3.7 -m pip install tflite_runtime-2.1.0.post1-cp37-cp37m-linux_x86_64.whl \
|
|
|
|
|
&& rm tflite_runtime-2.1.0.post1-cp37-cp37m-linux_x86_64.whl \
|
2020-01-05 17:43:14 -06:00
|
|
|
&& rm -rf /var/lib/apt/lists/* \
|
|
|
|
|
&& (apt-get autoremove -y; apt-get autoclean -y)
|
2019-05-27 10:17:57 -05:00
|
|
|
|
2020-02-18 05:55:06 -06:00
|
|
|
# get model and labels
|
2020-07-26 12:03:44 -05:00
|
|
|
RUN wget -q https://github.com/google-coral/edgetpu/raw/master/test_data/ssd_mobilenet_v2_coco_quant_postprocess_edgetpu.tflite -O /edgetpu_model.tflite --trust-server-names
|
2020-09-07 12:17:42 -05:00
|
|
|
COPY labelmap.txt /labelmap.txt
|
2020-07-26 12:03:44 -05:00
|
|
|
RUN wget -q https://github.com/google-coral/edgetpu/raw/master/test_data/ssd_mobilenet_v2_coco_quant_postprocess.tflite -O /cpu_model.tflite
|
2020-07-02 14:43:33 +01:00
|
|
|
|
2019-03-17 09:03:52 -05:00
|
|
|
|
2020-07-26 07:50:30 -05:00
|
|
|
RUN mkdir /cache /clips
|
2020-07-09 06:57:16 -05:00
|
|
|
|
2019-02-28 06:49:27 -06:00
|
|
|
WORKDIR /opt/frigate/
|
|
|
|
|
ADD frigate frigate/
|
2019-01-26 08:02:59 -06:00
|
|
|
COPY detect_objects.py .
|
2020-02-22 08:59:16 -06:00
|
|
|
COPY benchmark.py .
|
2019-01-26 08:02:59 -06:00
|
|
|
|
2020-02-18 05:55:06 -06:00
|
|
|
CMD ["python3.7", "-u", "detect_objects.py"]
|
