Skip to content

用python语言以C++的风格重写SSD的后处理,caffe、onnx、tensorRT三个版本

Notifications You must be signed in to change notification settings

cqu20160901/ssd_caffe_onnx_tensorRT

Repository files navigation

ssd 检测模型移植

用python语言,以C语言的形式重写了SSD的后处理,便于移植不同平台。caffe、onnx、tensorRT 三种方式测试结果

文件夹结构说明:

caffe_ssd_original:原始的ssd_caffe对应prototxt、caffeModel、测试图像、测试结果、测试demo脚本

caffe_ssd_transplant:去除维度变换层的prototxt、caffeModel、测试图像、测试结果、测试demo脚本

onnx_ssd_transplant:onnx模型、测试图像、测试结果、测试demo脚本

tensorRT_ssd_transplant:TensorRT版本模型、测试图像、测试结果、测试demo脚本、onnx模型、onnx2tensorRT脚本(tensorRT-7.2.3.4)

测试结果

移植后处理结果

image

About

用python语言以C++的风格重写SSD的后处理,caffe、onnx、tensorRT三个版本

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages