TensorRT-SSD
TensorRT-SSD copied to clipboard
The confidence of the prediction will exceed 1 after addding extra prior box
I changed the number of categories(30) and the proportion of boxes(add 3 aspect ratio in the 4_3 layer,and 4 in the fc7 and conv6_2 layer). I changed all the 21 to 30 in pluginImplement.cpp and pluginImplement.h. And changed the appropriate number is in cudaSoftmax. When I predict, the precision is very low. And the confidence level is greater than 1. What are the reasons and what else do I need to modify?
You should modify the aspect ratio parameter of PriorBox Layer in pluginImplement.cpp
Yes, I changed it.
I find that it is not a question of prior box, but a question of category. I used SSD to train my own dataset, and the results were not correct.