zishuo/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2025-09-26 19:41:29 +08:00
Code Issues Packages Projects Releases Wiki Activity

Implement API to train classification models (#18475)

Browse Source
This commit is contained in:
Nicolas Mowen
2025-05-29 17:51:32 -06:00
committed by Blake Blackshear
parent 6dc36fcbb4
commit 2c7b71b16e
8 changed files with 219 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
docker/main/Dockerfile
View File

@@ -227,6 +227,9 @@ ENV OPENCV_FFMPEG_LOGLEVEL=8
# Set HailoRT to disable logging
ENV HAILORT_LOGGER_PATH=NONE
# TensorFlow error only
ENV TF_CPP_MIN_LOG_LEVEL=3
ENV PATH="/usr/local/go2rtc/bin:/usr/local/tempio/bin:/usr/local/nginx/sbin:${PATH}"
# Install dependencies

3
docker/main/requirements-wheels.txt
View File

@@ -11,6 +11,9 @@ joserfc == 1.0.*
pathvalidate == 3.2.*
markupsafe == 3.0.*
python-multipart == 0.0.12
# Classification Model Training
tensorflow == 2.19.* ; platform_machine == 'aarch64'
tensorflow-cpu == 2.19.* ; platform_machine == 'x86_64'
# General
mypy == 1.6.1
onvif-zeep-async == 3.1.*