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

docker/build_pycuda.sh

@@ -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/*

docker/install_tensorrt.sh

@@ -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/*

docker/tensorrt_models.sh

@@ -31,4 +31,7 @@ do
     python3 yolo_to_onnx.py -m ${model}
     python3 onnx_to_tensorrt.py -m ${model}
     cp /tensorrt_demos/yolo/${model}.trt ${OUTPUT_FOLDER}/${model}.trt;
-done
+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

docs/docs/configuration/detectors.md

@@ -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
+```