mirror of
https://github.com/blakeblackshear/frigate.git
synced 2026-02-02 09:15:22 +03:00
16 lines
422 B
Bash
Executable File
16 lines
422 B
Bash
Executable File
#!/bin/bash
|
|
|
|
set -xe
|
|
cd /tensorrt_demos/plugins && make
|
|
cp /tensorrt_demos/plugins/libyolo_layer.so /plugin/libyolo_layer.so
|
|
|
|
cd /tensorrt_demos/yolo
|
|
for model in yolov4-tiny-288 \
|
|
yolov4-tiny-416 \
|
|
yolov4-288 \
|
|
yolov4-416 ; do
|
|
python3 yolo_to_onnx.py -m ${model}
|
|
python3 onnx_to_tensorrt.py -m ${model}
|
|
cp /tensorrt_demos/yolo/${model}.trt /model/${model}.trt
|
|
done
|