I have no idea how to give the prior box parameters such as minSize, maxSize, aspectRatios, numMinSize, numMaxSize and numAspectRatios. For example, defferent PriorBox layers may have different minSizes, and one PriorBox may have two aspect ratios, how could I pass these parameters when I creat PriorBox plugin layer? Could someone give me some suggestions? Thank you in advance!
@fujiaweigege For each priorbox layer,you should feed the parameters which in prototxt. So, you should do like this
float min_size=60.0, max_size=110.0, aspect_ratio[3]={1.0, 2.0, 3.0};
mPluginPriorBoxfc7 = std::unique_ptr<INvPlugin, decltype(nvPluginDeleter)>
(createSSDPriorBoxPlugin({&min_size, &max_size, aspect_ratio, 1 , 1, 3, true, false, {0.1, 0.1, 0.2, 0.2}, 0, 0, 16.0, 16.0, 0.5}), nvPluginDeleter); You can do this more gentle.
When I use the createSSDDetectionOutputPlugin plugin, I have the same problem about “Plugin layer output count is not equal to caffe output count”.However, when I remove the detectout layer from prototxt file, it can run well and I can get output of the mbox_loc and mbox_conf layer