(custom layers implementation) Traffic Light Detection

[JerrySciangula]

Hi everyone,

I was trying to understand the neural network Traffic Light Detection in the Perception module.

Reading its prototxt file, i found the custom RCNNProposal and DFMBPSROIAlign layers.

Could someone explain me how they work or load the code that implemented them?

Thanks in advance.

[jeroldchen]

@JerrySciangula We have uploaded TensorRT implementation for custom plugins of traffic light detection in recent master code. You can check this: https://github.com/ApolloAuto/apollo/commit/a916d039779127b2b15cb252e5fbc6e022624be2 .

[hawl666]

@jeroldchen Is it possible to use this TensorRT plugin to write python code for traffic signal detection?