mirror of
https://github.com/blakeblackshear/frigate.git
synced 2024-10-15 20:07:23 +02:00
4a35573210
* Initial support for Hailo-8L Added file for Hailo-8L detector including dockerfile, h8l.mk, h8l.hcl, hailo8l.py, ci.yml and ssd_mobilenat_v1.hef as the inference network. Added files to help with the installation of Hailo-8L dependences like generate_wheel_conf.py, requirements-wheel-h8l.txt and modified setup.py to try and work with any hardware. Updated docs to reflect Initial Hailo-8L support including oject_detectors.md, hardware.md and installation.md. * Update .github/workflows/ci.yml typo h8l not arm64 Co-authored-by: Nicolas Mowen <nickmowen213@gmail.com> * Update docs/docs/configuration/object_detectors.md Clarity for the end user and correct uses of words Co-authored-by: Nicolas Mowen <nickmowen213@gmail.com> * Update docs/docs/frigate/installation.md typo Co-authored-by: Nicolas Mowen <nickmowen213@gmail.com> * update Installation.md to clarify Hailo-8L installation process. * Update docs/docs/frigate/hardware.md Co-authored-by: Josh Hawkins <32435876+hawkeye217@users.noreply.github.com> * Update hardware.md add Inference time. * Oops no new line at the end of the file. * Update docs/docs/frigate/hardware.md typo Co-authored-by: Josh Hawkins <32435876+hawkeye217@users.noreply.github.com> * Update dockerfile to download the ssd_modilenet_v1 model instead of having it in the repo. * Updated dockerfile so it dose not download the model file. add function to download it at runtime. update model path. * fix formatting according to ruff and removed unnecessary functions. --------- Co-authored-by: Nicolas Mowen <nickmowen213@gmail.com> Co-authored-by: Josh Hawkins <32435876+hawkeye217@users.noreply.github.com>
101 lines
3.9 KiB
Docker
101 lines
3.9 KiB
Docker
# syntax=docker/dockerfile:1.6
|
|
|
|
ARG DEBIAN_FRONTEND=noninteractive
|
|
|
|
# Build Python wheels
|
|
FROM wheels AS h8l-wheels
|
|
|
|
COPY docker/main/requirements-wheels.txt /requirements-wheels.txt
|
|
COPY docker/hailo8l/requirements-wheels-h8l.txt /requirements-wheels-h8l.txt
|
|
|
|
RUN sed -i "/https:\/\//d" /requirements-wheels.txt
|
|
|
|
# Create a directory to store the built wheels
|
|
RUN mkdir /h8l-wheels
|
|
|
|
# Build the wheels
|
|
RUN pip3 wheel --wheel-dir=/h8l-wheels -c /requirements-wheels.txt -r /requirements-wheels-h8l.txt
|
|
|
|
# Build HailoRT and create wheel
|
|
FROM deps AS build-hailort
|
|
|
|
# Install necessary APT packages
|
|
RUN apt-get update && apt-get install -y \
|
|
build-essential \
|
|
cmake \
|
|
git \
|
|
wget \
|
|
python3-dev \
|
|
gcc-9 \
|
|
g++-9 \
|
|
libzmq3-dev \
|
|
pciutils \
|
|
rsync \
|
|
&& rm -rf /var/lib/apt/lists/*
|
|
|
|
# Extract Python version and set environment variables
|
|
RUN PYTHON_VERSION=$(python3 --version 2>&1 | awk '{print $2}' | cut -d. -f1,2) && \
|
|
PYTHON_VERSION_NO_DOT=$(echo $PYTHON_VERSION | sed 's/\.//') && \
|
|
echo "PYTHON_VERSION=$PYTHON_VERSION" > /etc/environment && \
|
|
echo "PYTHON_VERSION_NO_DOT=$PYTHON_VERSION_NO_DOT" >> /etc/environment
|
|
|
|
#ENV PYTHON_VER=$PYTHON_VERSION
|
|
#ENV PYTHON_VER_NO_DOT=$PYTHON_VERSION_NO_DOT
|
|
|
|
# Clone and build HailoRT
|
|
RUN . /etc/environment && \
|
|
git clone https://github.com/hailo-ai/hailort.git /opt/hailort && \
|
|
cd /opt/hailort && \
|
|
git checkout v4.17.0 && \
|
|
cmake -H. -Bbuild -DCMAKE_BUILD_TYPE=Release -DHAILO_BUILD_PYBIND=1 -DPYBIND11_PYTHON_VERSION=${PYTHON_VERSION} && \
|
|
cmake --build build --config release --target libhailort && \
|
|
cmake --build build --config release --target _pyhailort && \
|
|
cp build/hailort/libhailort/bindings/python/src/_pyhailort.cpython-${PYTHON_VERSION_NO_DOT}-aarch64-linux-gnu.so hailort/libhailort/bindings/python/platform/hailo_platform/pyhailort/ && \
|
|
cp build/hailort/libhailort/src/libhailort.so hailort/libhailort/bindings/python/platform/hailo_platform/pyhailort/
|
|
|
|
RUN ls -ahl /opt/hailort/build/hailort/libhailort/src/
|
|
RUN ls -ahl /opt/hailort/hailort/libhailort/bindings/python/platform/hailo_platform/pyhailort/
|
|
|
|
# Remove the existing setup.py if it exists in the target directory
|
|
RUN rm -f /opt/hailort/hailort/libhailort/bindings/python/platform/setup.py
|
|
|
|
# Copy generate_wheel_conf.py and setup.py
|
|
COPY docker/hailo8l/pyhailort_build_scripts/generate_wheel_conf.py /opt/hailort/hailort/libhailort/bindings/python/platform/generate_wheel_conf.py
|
|
COPY docker/hailo8l/pyhailort_build_scripts/setup.py /opt/hailort/hailort/libhailort/bindings/python/platform/setup.py
|
|
|
|
# Run the generate_wheel_conf.py script
|
|
RUN python3 /opt/hailort/hailort/libhailort/bindings/python/platform/generate_wheel_conf.py
|
|
|
|
# Create a wheel file using pip3 wheel
|
|
RUN cd /opt/hailort/hailort/libhailort/bindings/python/platform && \
|
|
python3 setup.py bdist_wheel --dist-dir /hailo-wheels
|
|
|
|
# Use deps as the base image
|
|
FROM deps AS h8l-frigate
|
|
|
|
# Copy the wheels from the wheels stage
|
|
COPY --from=h8l-wheels /h8l-wheels /deps/h8l-wheels
|
|
COPY --from=build-hailort /hailo-wheels /deps/hailo-wheels
|
|
COPY --from=build-hailort /etc/environment /etc/environment
|
|
RUN CC=$(python3 -c "import sysconfig; import shlex; cc = sysconfig.get_config_var('CC'); cc_cmd = shlex.split(cc)[0]; print(cc_cmd[:-4] if cc_cmd.endswith('-gcc') else cc_cmd)") && \
|
|
echo "CC=$CC" >> /etc/environment
|
|
|
|
# Install the wheels
|
|
RUN pip3 install -U /deps/h8l-wheels/*.whl
|
|
RUN pip3 install -U /deps/hailo-wheels/*.whl
|
|
|
|
RUN . /etc/environment && \
|
|
mv /usr/local/lib/python${PYTHON_VERSION}/dist-packages/hailo_platform/pyhailort/libhailort.so /usr/lib/${CC} && \
|
|
cd /usr/lib/${CC}/ && \
|
|
ln -s libhailort.so libhailort.so.4.17.0
|
|
|
|
# Copy base files from the rootfs stage
|
|
COPY --from=rootfs / /
|
|
|
|
# Set environment variables for Hailo SDK
|
|
ENV PATH="/opt/hailort/bin:${PATH}"
|
|
ENV LD_LIBRARY_PATH="/usr/lib/aarch64-linux-gnu:${LD_LIBRARY_PATH}"
|
|
|
|
# Set workdir
|
|
WORKDIR /opt/frigate/
|