I am trying to implement SSD on TensorRT 5.0 with Resnet as backbone network for feature extraction. Since the model implementation is in Tensorflow, I had to use UFF converter with a preprocessor , config.py for conversion. The sample example by nvidia on Inception SSD was taken as reference for the same and the config.py was modified in order to handle the changes in the network.
However I am stuck at the SSD Prior box implementation. Here are some of my questions:
How is GridAnchoGenerator plugin different from PriorBox Plugin ? Which one should be used with SSD?
In my current implementation I have used PriorBox Plugin with 6 intermediate feature maps in order to generate the anchor boxes and while parsing the network , I am getting the following error :
priorBoxPlugin.cpp:91: virtual nvinfer1::Dims nvinfer1::plugin::PriorBox::getOutputDimensions(int, const nvinfer1::Dims*, int): Assertion `nbInputDims == 2’ failed
Any help on the topic would be very useful. Thanks in advance.