I have a convolution layer which was preceded by a padding layer(custom plugin).
input placeholder : 1×3x361×641
pad (custom layer) : 1×3x367×647
conv out expected : 1×32x181×321
Note : pad custom plugin done as there is issue in tensorrt pad as shared in previous topic
I have verified the custom layer output and observed works as expected.
i have tried using
Dims getOutputDimensions(int index, const Dims* inputs, int nbInputDims) override
{
option a) tried
return DimsNCHW(1,inputs[0].d[0], inputs[0].d[1] + 2 * hPad, inputs[0].d[2] + 2 * wPad);
// wPad and hPad is 3
and using of
option b) tried
return DimsCHW(inputs[0].d[0], inputs[0].d[1] + 2 * hPad, inputs[0].d[2] + 2 * wPad);
}
error obtained option a)
Parser error: SharedFeatureExtractor/resnet_v1_50/conv1/Conv2D: Order size is not matching the number dimensions of TensorRT
error obtianed for option b)
ERROR: SharedFeatureExtractor/resnet_v1_50/conv1/Conv2D: image size is smaller than filter size
ERROR: UFFParser: Parser error: MarkOutput_0: Order size is not matching the number dimensions of TensorRT
however the conv filter is 7x7x3x32 in tensorflow
any idea why this error is obtained.