Compare commits

..

2 Commits

Author SHA1 Message Date
Rémi Bédard-Couture 55175f9b31
Update default config to leverage openvino (#2763) 2024-04-08 21:41:54 -04:00
Rémi Bédard-Couture 1cded8c053
Fix openvino model generation in Frigate install script (#2762)
* get pip from APT

* Add openvino model conversion to make it readily available
2024-04-08 21:21:08 -04:00
1 changed files with 14 additions and 5 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_info "Installing Python3 Dependencies"
$STD apt-get install -y {python3,python3-dev,python3-setuptools,python3-distutils}
wget -q https://bootstrap.pypa.io/get-pip.py -O get-pip.py
$STD python3 get-pip.py --quiet "pip"
$STD apt-get install -y {python3,python3-dev,python3-setuptools,python3-distutils,python3-pip}
msg_ok "Installed Python3 Dependencies"
msg_info "Installing Node.js"
@ -79,8 +77,18 @@ sed -i '/^s6-svc -O \.$/s/^/#/' /opt/frigate/docker/main/rootfs/etc/s6-overlay/s
cat <<EOF >/config/config.yml
mqtt:
enabled: false
detectors:
ov:
type: openvino
device: AUTO
model:
path: /openvino-model/FP16/ssdlite_mobilenet_v2.xml
model:
path: /cpu_model.tflite
width: 300
height: 300
input_tensor: nhwc
input_pixel_format: bgr
labelmap_path: /openvino-model/coco_91cl_bkgr.txt
cameras:
test:
ffmpeg:
@ -104,7 +112,8 @@ msg_info "Installing Object Detection Models (Resilience)"
$STD pip install -r /opt/frigate/docker/main/requirements-ov.txt
cd /opt/frigate/models
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 ..
export CCACHE_DIR=/root/.ccache
export CCACHE_MAXSIZE=2G