|
|
|
|
@@ -146,7 +146,7 @@ ldconfig
|
|
|
|
|
msg_ok "Built libUSB"
|
|
|
|
|
|
|
|
|
|
msg_info "Bootstrapping pip"
|
|
|
|
|
curl_with_retry "https://bootstrap.pypa.io/get-pip.py" "/tmp/get-pip.py"
|
|
|
|
|
wget -q https://bootstrap.pypa.io/get-pip.py -O /tmp/get-pip.py
|
|
|
|
|
sed -i 's/args.append("setuptools")/args.append("setuptools==77.0.3")/' /tmp/get-pip.py
|
|
|
|
|
$STD python3 /tmp/get-pip.py "pip"
|
|
|
|
|
rm -f /tmp/get-pip.py
|
|
|
|
|
@@ -169,13 +169,13 @@ NODE_VERSION="20" setup_nodejs
|
|
|
|
|
|
|
|
|
|
msg_info "Downloading Inference Models"
|
|
|
|
|
mkdir -p /models /openvino-model
|
|
|
|
|
curl_with_retry "https://github.com/google-coral/test_data/raw/release-frogfish/ssdlite_mobiledet_coco_qat_postprocess_edgetpu.tflite" "/edgetpu_model.tflite"
|
|
|
|
|
curl_with_retry "https://github.com/google-coral/test_data/raw/release-frogfish/ssdlite_mobiledet_coco_qat_postprocess.tflite" "/models/cpu_model.tflite"
|
|
|
|
|
wget -q -O /edgetpu_model.tflite https://github.com/google-coral/test_data/raw/release-frogfish/ssdlite_mobiledet_coco_qat_postprocess_edgetpu.tflite
|
|
|
|
|
wget -q -O /models/cpu_model.tflite https://github.com/google-coral/test_data/raw/release-frogfish/ssdlite_mobiledet_coco_qat_postprocess.tflite
|
|
|
|
|
cp /opt/frigate/labelmap.txt /labelmap.txt
|
|
|
|
|
msg_ok "Downloaded Inference Models"
|
|
|
|
|
|
|
|
|
|
msg_info "Downloading Audio Model"
|
|
|
|
|
curl_with_retry "https://www.kaggle.com/api/v1/models/google/yamnet/tfLite/classification-tflite/1/download" "/tmp/yamnet.tar.gz"
|
|
|
|
|
wget -q -O /tmp/yamnet.tar.gz https://www.kaggle.com/api/v1/models/google/yamnet/tfLite/classification-tflite/1/download
|
|
|
|
|
$STD tar xzf /tmp/yamnet.tar.gz -C /
|
|
|
|
|
mv /1.tflite /cpu_audio_model.tflite
|
|
|
|
|
cp /opt/frigate/audio-labelmap.txt /audio-labelmap.txt
|
|
|
|
|
@@ -205,7 +205,7 @@ msg_ok "Installed OpenVino"
|
|
|
|
|
|
|
|
|
|
msg_info "Building OpenVino Model"
|
|
|
|
|
cd /models
|
|
|
|
|
curl_with_retry "http://download.tensorflow.org/models/object_detection/ssdlite_mobilenet_v2_coco_2018_05_09.tar.gz" "ssdlite_mobilenet_v2_coco_2018_05_09.tar.gz"
|
|
|
|
|
wget -q http://download.tensorflow.org/models/object_detection/ssdlite_mobilenet_v2_coco_2018_05_09.tar.gz
|
|
|
|
|
$STD tar -zxf ssdlite_mobilenet_v2_coco_2018_05_09.tar.gz --no-same-owner
|
|
|
|
|
if python3 /opt/frigate/docker/main/build_ov_model.py &>/dev/null; then
|
|
|
|
|
mkdir -p /openvino-model
|
|
|
|
|
@@ -219,7 +219,7 @@ if python3 /opt/frigate/docker/main/build_ov_model.py &>/dev/null; then
|
|
|
|
|
if [[ -n "$OV_LABELS" ]]; then
|
|
|
|
|
ln -sf "$OV_LABELS" /openvino-model/coco_91cl_bkgr.txt
|
|
|
|
|
else
|
|
|
|
|
curl_with_retry "https://raw.githubusercontent.com/openvinotoolkit/open_model_zoo/master/data/dataset_classes/coco_91cl_bkgr.txt" "/openvino-model/coco_91cl_bkgr.txt"
|
|
|
|
|
wget -q "https://raw.githubusercontent.com/openvinotoolkit/open_model_zoo/master/data/dataset_classes/coco_91cl_bkgr.txt" -O /openvino-model/coco_91cl_bkgr.txt
|
|
|
|
|
fi
|
|
|
|
|
fi
|
|
|
|
|
sed -i 's/truck/car/g' /openvino-model/coco_91cl_bkgr.txt
|
|
|
|
|
@@ -246,7 +246,7 @@ msg_info "Configuring Frigate"
|
|
|
|
|
mkdir -p /config /media/frigate
|
|
|
|
|
cp -r /opt/frigate/config/. /config
|
|
|
|
|
|
|
|
|
|
curl_with_retry "https://github.com/intel-iot-devkit/sample-videos/raw/master/person-bicycle-car-detection.mp4" "/media/frigate/person-bicycle-car-detection.mp4"
|
|
|
|
|
curl -fsSL "https://github.com/intel-iot-devkit/sample-videos/raw/master/person-bicycle-car-detection.mp4" -o "/media/frigate/person-bicycle-car-detection.mp4"
|
|
|
|
|
|
|
|
|
|
echo "tmpfs /tmp/cache tmpfs defaults 0 0" >>/etc/fstab
|
|
|
|
|
|
|
|
|
|
@@ -289,7 +289,7 @@ detect:
|
|
|
|
|
enabled: false
|
|
|
|
|
EOF
|
|
|
|
|
|
|
|
|
|
if grep -q -o -m1 -E 'avx[^ ]*|sse4_2' /proc/cpuinfo && [[ -f /openvino-model/ssdlite_mobilenet_v2.xml ]] && [[ -f /openvino-model/coco_91cl_bkgr.txt ]]; then
|
|
|
|
|
if grep -q -o -m1 -E 'avx[^ ]*|sse4_2' /proc/cpuinfo; then
|
|
|
|
|
cat <<EOF >>/config/config.yml
|
|
|
|
|
ffmpeg:
|
|
|
|
|
hwaccel_args: auto
|
|
|
|
|
|