经过大约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)。请帮助我。