Fix openvino model generation in Frigate install script (#2762)

* get pip from APT

* Add openvino model conversion to make it readily available
This commit is contained in:
Rémi Bédard-Couture 2024-04-08 21:21:08 -04:00 committed by GitHub
parent 2847d9cc9a
commit 1cded8c053
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 4 deletions

View File

@ -18,9 +18,7 @@ $STD apt-get install -y {curl,sudo,mc,git,gpg,automake,build-essential,xz-utils,
msg_ok "Installed Dependencies" msg_ok "Installed Dependencies"
msg_info "Installing Python3 Dependencies" msg_info "Installing Python3 Dependencies"
$STD apt-get install -y {python3,python3-dev,python3-setuptools,python3-distutils} $STD apt-get install -y {python3,python3-dev,python3-setuptools,python3-distutils,python3-pip}
wget -q https://bootstrap.pypa.io/get-pip.py -O get-pip.py
$STD python3 get-pip.py --quiet "pip"
msg_ok "Installed Python3 Dependencies" msg_ok "Installed Python3 Dependencies"
msg_info "Installing Node.js" msg_info "Installing Node.js"
@ -104,7 +102,8 @@ msg_info "Installing Object Detection Models (Resilience)"
$STD pip install -r /opt/frigate/docker/main/requirements-ov.txt $STD pip install -r /opt/frigate/docker/main/requirements-ov.txt
cd /opt/frigate/models cd /opt/frigate/models
export ENABLE_ANALYTICS=NO export ENABLE_ANALYTICS=NO
$STD /usr/local/bin/omz_downloader --name ssdlite_mobilenet_v2 $STD /usr/local/bin/omz_downloader --name ssdlite_mobilenet_v2 --num_attempts 2
$STD /usr/local/bin/omz_converter --name ssdlite_mobilenet_v2 --precision FP16 --mo /usr/local/bin/mo
cd .. cd ..
export CCACHE_DIR=/root/.ccache export CCACHE_DIR=/root/.ccache
export CCACHE_MAXSIZE=2G export CCACHE_MAXSIZE=2G