Implement R-FCN in TensorRT,how to write the psroi_pooling layer(customer layer----pluginImplement).

Robotics & Edge Computing Jetson & Embedded Systems Jetson TX2
1

Hi,I want to know how to implement R-FCN in TensorRT.And anyone who can give some samples about how to write the psroi_pooling layer(customer layer----pluginImplement).How can I implement the customer layer (psroi_pooling layer)of the deploy.prototxt?Any advice.

2

There is a sample implementation of FasterRCNN, in there they create a custom pluging called “RPROIFused”, I think you can use that as a baseline to implement psroi_pooling.
the sample is under

tensorrt/samples/sampleFasterRCNN
3

Thanks