TensorRT-SSD icon indicating copy to clipboard operation
TensorRT-SSD copied to clipboard

The confidence of the prediction will exceed 1 after addding extra prior box

Open 408550969 opened this issue 6 years ago • 3 comments

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?

408550969 avatar Aug 03 '18 01:08 408550969

You should modify the aspect ratio parameter of PriorBox Layer in pluginImplement.cpp

chenzhi1992 avatar Aug 04 '18 05:08 chenzhi1992

Yes, I changed it.

408550969 avatar Aug 04 '18 05:08 408550969

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.

408550969 avatar Aug 04 '18 07:08 408550969