mirror of
https://github.com/blakeblackshear/frigate.git
synced 2024-11-30 19:09:13 +01:00
7522bb6fab
* support for other yolov models and config checks * apply code formatting * Information about core mask and inference speed * update rknn postprocess and remove params * update model selection * Apply suggestions from code review Co-authored-by: Nicolas Mowen <nickmowen213@gmail.com> --------- Co-authored-by: Nicolas Mowen <nickmowen213@gmail.com>
34 lines
1.5 KiB
Docker
34 lines
1.5 KiB
Docker
# syntax=docker/dockerfile:1.6
|
|
|
|
# https://askubuntu.com/questions/972516/debian-frontend-environment-variable
|
|
ARG DEBIAN_FRONTEND=noninteractive
|
|
|
|
FROM wheels as rk-wheels
|
|
COPY docker/main/requirements-wheels.txt /requirements-wheels.txt
|
|
COPY docker/rockchip/requirements-wheels-rk.txt /requirements-wheels-rk.txt
|
|
RUN sed -i "/https:\/\//d" /requirements-wheels.txt
|
|
RUN pip3 wheel --wheel-dir=/rk-wheels -c /requirements-wheels.txt -r /requirements-wheels-rk.txt
|
|
|
|
FROM wget as rk-downloads
|
|
RUN wget -qO librknnrt.so https://github.com/MarcA711/rknpu2/raw/master/runtime/RK3588/Linux/librknn_api/aarch64/librknnrt.so
|
|
RUN wget -qO ffmpeg https://github.com/MarcA711/Rockchip-FFmpeg-Builds/releases/download/latest/ffmpeg
|
|
RUN wget -qO ffprobe https://github.com/MarcA711/Rockchip-FFmpeg-Builds/releases/download/latest/ffprobe
|
|
RUN wget -qO yolov8n-320x320.rknn https://github.com/MarcA711/rknn-models/releases/download/latest/yolov8n-320x320.rknn
|
|
FROM deps AS rk-deps
|
|
ARG TARGETARCH
|
|
|
|
RUN --mount=type=bind,from=rk-wheels,source=/rk-wheels,target=/deps/rk-wheels \
|
|
pip3 install -U /deps/rk-wheels/*.whl
|
|
|
|
WORKDIR /opt/frigate/
|
|
COPY --from=rootfs / /
|
|
COPY --from=rk-downloads /rootfs/librknnrt.so /usr/lib/
|
|
COPY --from=rk-downloads /rootfs/yolov8n-320x320.rknn /models/
|
|
|
|
RUN rm -rf /usr/lib/btbn-ffmpeg/bin/ffmpeg
|
|
RUN rm -rf /usr/lib/btbn-ffmpeg/bin/ffprobe
|
|
COPY --from=rk-downloads /rootfs/ffmpeg /usr/lib/btbn-ffmpeg/bin/
|
|
COPY --from=rk-downloads /rootfs/ffprobe /usr/lib/btbn-ffmpeg/bin/
|
|
RUN chmod +x /usr/lib/btbn-ffmpeg/bin/ffmpeg
|
|
RUN chmod +x /usr/lib/btbn-ffmpeg/bin/ffprobe
|