上传者: 42181319
|
上传时间: 2022-03-12 16:47:34
|
文件大小: 441KB
|
文件类型: -
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