-
将yolov3-tiny模型通过onnx转换为TensorRT的trt模型
资源介绍
yolov3-tiny2onnx2trt
将您的yolov3-tiny模型转换为trt模型
设备：nvidia jetson tx2
jetpack版本：jetpack4.2：
ubuntu18.04
tensorrt5.0.6.3
cuda10.0
cudnn7.3.1
其他：
python=2.7
numpy=1.16.1
onnx=1.4.1 (important)
pycuda=2019.1.1
Pillow=6.1.0
wget=3.2
自定义设置
data_processing.py:
line14: LABEL_FILE_PATH = '/home/nvidia/yolov3-tiny2onnx2trt/coco_labels.txt'
line19: CATEGORY_NUM = 80
yolov3_to_onnx.py:
line778: img_siz