I’m trying to write a priorbox layer in SSD for driveworks plugin, and the priorbox layer need to use the PriorBoxParameters from model.prototxt. However, in driveworks, the constructor of a plugin layer seems directly load from weights, as follows,
Did you check your model working well with TensorRT first? Including plugin layer code in DW is similar to including plugin layer in TensorRT. Please check TensorRT sample_plugin and dnn_plugin in DW as a reference.
_dwDNNPlugin_initialize() function can read plugin information byte by byte as shown in sample.
Yes, I just write the driveworks plugin class with reference to tensorRT.
However, the priorBOX layer is constructed by the deploy file under caffe model in tensorRT. The API of Driveworks need weights to construct the plugin layer. It’s totally different. PriorBox doesn’t need weights.
So I have no idea how to modify it from tensorRT plugin class to driveworks plugin class.