1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162 |
- FROM ubuntu:20.04
- LABEL maintainer "blakeb@blakeshome.com"
- ENV DEBIAN_FRONTEND=noninteractive
- # Install packages for apt repo
- RUN apt -qq update && apt -qq install --no-install-recommends -y \
- software-properties-common \
- # apt-transport-https ca-certificates \
- build-essential \
- gnupg wget unzip tzdata \
- # libcap-dev \
- && add-apt-repository ppa:deadsnakes/ppa -y \
- && apt -qq install --no-install-recommends -y \
- python3.8 \
- python3.8-dev \
- python3-pip \
- ffmpeg \
- # VAAPI drivers for Intel hardware accel
- libva-drm2 libva2 i965-va-driver vainfo \
- && python3.8 -m pip install -U pip \
- && python3.8 -m pip install -U wheel setuptools \
- && python3.8 -m pip install -U \
- opencv-python-headless \
- # python-prctl \
- numpy \
- imutils \
- scipy \
- psutil \
- && python3.8 -m pip install -U \
- Flask \
- paho-mqtt \
- PyYAML \
- matplotlib \
- click \
- && 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 \
- libedgetpu1-max \
- ## Tensorflow lite
- && wget -q https://dl.google.com/coral/python/tflite_runtime-2.1.0.post1-cp38-cp38-linux_x86_64.whl \
- && python3.8 -m pip install tflite_runtime-2.1.0.post1-cp38-cp38-linux_x86_64.whl \
- && rm tflite_runtime-2.1.0.post1-cp38-cp38-linux_x86_64.whl \
- && rm -rf /var/lib/apt/lists/* \
- && (apt-get autoremove -y; apt-get autoclean -y)
- # get model and labels
- 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
- COPY labelmap.txt /labelmap.txt
- RUN wget -q https://github.com/google-coral/edgetpu/raw/master/test_data/ssd_mobilenet_v2_coco_quant_postprocess.tflite -O /cpu_model.tflite
- RUN mkdir /cache /clips
- WORKDIR /opt/frigate/
- ADD frigate frigate/
- COPY detect_objects.py .
- COPY benchmark.py .
- COPY process_clip.py .
- CMD ["python3.8", "-u", "detect_objects.py"]
|