使用TensorRT插件使用快速RCNN进行自定义尺寸图像的对象检测问题

时间:2019-06-29 07:21:42

标签: tensorrt faster-rcnn

经过大约2天的努力,我提出了这个问题,但仍无法解决我现有的问题。我有一个 5000 * 600 的图像,我想使用Tensorrt插件使用Faster RCNN进行推断。

更快的R-CNN将3通道375x500图像作为输入from the official documentation。我从file看到了相同的维度。输入行号 26和27 。我试图更改行号 26和27 中的尺寸,以便可以推断出自定义图像形状。但这没什么用。

我已经使用Nvidia的官方文档在Ubuntu系统中安装了 tensorrt ,并且正在/ usr / src / tensorrt中工作。有样本和数据。可以在路径/ usr / src / tensorrt / samples / sampleFasterRCNN中找到FasterRCNN的配置文件( sampleFasterRCNN.cpp )。

我正面临另一个问题。如果我要推断单个图像,即使用 imagelist 中的一幅图像。请参见file中的行号294(std :: vector imageList = {“ 004545.ppm”,“ test12.ppm”};),它显示了核心转储时的错误。

我没有传递单个图像,而是传递了多个图像,这样的错误(核心转储)得以解决。我不明白为什么?

我想使用tensorrt插件使用自定义图像形状自定义Faster RCNN。有没有办法做到这一点。我正在使用来自TensorFlow Zoo的冻结文件(.pb)。请帮助我。

0 个答案:

没有答案