mirror of
https://github.com/blakeblackshear/frigate.git
synced 2024-12-23 19:11:14 +01:00
a96a951e23
* Non-Jetson changes Required for later commits: - Allow base image to be overridden (and don't assume its WORKDIR) - Ensure python3.9 - Map hwaccel decode presets as strings instead of lists Not required: - Fix existing documentation - Simplify hwaccel scale logic * Prepare for multi-arch tensorrt build * Add tensorrt images for Jetson boards * Add Jetson ffmpeg hwaccel * Update docs * Add CODEOWNERS * CI * Change default model from yolov7-tiny-416 to yolov7-320 In my experience the tiny models perform markedly worse without being much faster * fixup! Update docs
22 lines
572 B
Bash
Executable File
22 lines
572 B
Bash
Executable File
#!/bin/bash
|
|
|
|
set -euxo pipefail
|
|
|
|
SCRIPT_DIR="/usr/local/src/tensorrt_demos"
|
|
|
|
# Clone tensorrt_demos repo
|
|
git clone --depth 1 https://github.com/NateMeyer/tensorrt_demos.git -b conditional_download
|
|
|
|
# Build libyolo
|
|
if [ ! -e /usr/local/cuda ]; then
|
|
ln -s /usr/local/cuda-* /usr/local/cuda
|
|
fi
|
|
cd ./tensorrt_demos/plugins && make all -j$(nproc)
|
|
cp libyolo_layer.so /usr/local/lib/libyolo_layer.so
|
|
|
|
# Store yolo scripts for later conversion
|
|
cd ../
|
|
mkdir -p ${SCRIPT_DIR}/plugins
|
|
cp plugins/libyolo_layer.so ${SCRIPT_DIR}/plugins/libyolo_layer.so
|
|
cp -a yolo ${SCRIPT_DIR}/
|