administrator/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2026-02-03 09:45:22 +03:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

Add labelmap to trt model script and docs. Cleanup of old scripts.

Browse Source
This commit is contained in:
Nate Meyer 2022-12-28 12:55:34 -05:00
parent bd10a27376
commit c01507081e
4 changed files with 19 additions and 98 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

54
docker/build_pycuda.sh
View File

@ -1,54 +0,0 @@
#!/bin/bash
set -euxo pipefail
"${CUDA_LIB_VERSION:=11.8}"
"${CUDA_PKG_VERSION:=11-8}"
"${CUDNN_VERSION:=8.6.0.84}"
"${TENSORRT_VERSION:=7.2.2}"
# Add NVidia Repo
# wget -q -O - https://developer.download.nvidia.com/compute/cuda/repos/debian11/x86_64/3bf863cc.pub | apt-key add -
# echo "deb https://developer.download.nvidia.com/compute/cuda/repos/debian11/x86_64/ /" | tee /etc/apt/sources.list.d/nvidia-cuda.list
wget https://developer.download.nvidia.com/compute/cuda/repos/debian11/x86_64/cuda-keyring_1.0-1_all.deb
dpkg -i cuda-keyring_1.0-1_all.deb
add-apt-repository contrib
apt-get -qq update
# Utilities
apt-get install -y --no-install-recommends \
build-essential python3.9-dev python3-pip tar
python3 -m pip install --upgrade setuptools pip wheel numpy
# CUDA Base
apt-get install -y --no-install-recommends \
cuda-cudart-${CUDA_PKG_VERSION} \
cuda-compat-${CUDA_PKG_VERSION} \
cuda-cupti-${CUDA_PKG_VERSION} && \
ln -s cuda-${CUDA_LIB_VERSION} /usr/local/cuda
# CUDA Runtime
apt-get install -y --no-install-recommends \
cuda-libraries-${CUDA_PKG_VERSION} \
libnpp-${CUDA_PKG_VERSION} \
cuda-nvtx-${CUDA_PKG_VERSION} \
libcublas-${CUDA_PKG_VERSION}
# cuDNN Runtime
apt-get install -y --no-install-recommends \
libcudnn8
# Download pycuda source
wget https://github.com/inducer/pycuda/archive/refs/tags/v2022.2.tar.gz -O pycuda-v2022.2.tar.gz
tar xfz pycuda-v2022.2.tar.gz
# Build pycuda wheel
cd pycuda-v2022.2
python setup.py build
rm -rf /var/lib/apt/lists/*

43
docker/install_tensorrt.sh
View File

@ -1,43 +0,0 @@
#!/bin/bash
set -euxo pipefail
echo "${CUDA_LIB_VERSION:=11.7}"
echo "${CUDA_PKG_VERSION:=11-7}"
echo "${CUDNN_VERSION:=8.6.0.84}"
echo "${TENSORRT_VERSION:=8.4.1}"
# Add NVidia Repo
apt-get -qq update && apt-get install -y --no-install-recommends software-properties-common
# wget -q -O - https://developer.download.nvidia.com/compute/cuda/repos/debian11/x86_64/3bf863cc.pub | apt-key add -
# echo "deb https://developer.download.nvidia.com/compute/cuda/repos/debian11/x86_64/ /" | tee /etc/apt/sources.list.d/nvidia-cuda.list
wget https://developer.download.nvidia.com/compute/cuda/repos/debian11/x86_64/cuda-keyring_1.0-1_all.deb
dpkg -i cuda-keyring_1.0-1_all.deb
add-apt-repository contrib
apt-get -qq update
# CUDA Base
apt-get install -y --no-install-recommends \
cuda-cudart-${CUDA_PKG_VERSION} \
cuda-compat-${CUDA_PKG_VERSION} \
cuda-cupti-${CUDA_PKG_VERSION}
#ln -s cuda-${CUDA_LIB_VERSION} /usr/local/cuda
# CUDA Runtime
apt-get install -y --no-install-recommends \
cuda-libraries-${CUDA_PKG_VERSION} \
libnpp-${CUDA_PKG_VERSION} \
cuda-nvtx-${CUDA_PKG_VERSION} \
libcublas-${CUDA_PKG_VERSION}
# cuDNN Runtime
apt-get install -y --no-install-recommends \
libcudnn8
# TensorRT Runtime
apt-get install -y --no-install-recommends \
tensorrt-libs python3-libnvinfer
# apt-get clean autoclean -y
# apt-get autoremove --purge -y
# rm -rf /var/lib/apt/lists/*

3
docker/tensorrt_models.sh
View File

@ -32,3 +32,6 @@ do
python3 onnx_to_tensorrt.py -m ${model}
cp /tensorrt_demos/yolo/${model}.trt ${OUTPUT_FOLDER}/${model}.trt;
done
# Download Labelmap
wget -q https://github.com/openvinotoolkit/open_model_zoo/raw/master/data/dataset_classes/coco_91cl.txt -O ${OUTPUT_FOLDER}/coco_91cl.txt

15
docs/docs/configuration/detectors.md
View File

@ -202,3 +202,18 @@ yolov7-tiny-416
### Configuration Parameters
**TODO**
Sample:
```yaml
detectors:
tensorrt:
type: tensorrt
model:
path: /trt-models/yolov7-tiny-416.trt
labelmap_path: /trt-models/coco_91cl.txt
input_tensor: nchw
width: 416
height: 416
```