search

gaddekumar / tensorrt-ssd-easy

31 stars 26 forks source link

Why the confidence of the prediction will exceed 1? #14

Open 408550969 opened 6 years ago

408550969 commented 6 years ago

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? @JingliangGao @saikumarGadde

JingliangGao commented 6 years ago

which prototxt you used？

408550969 commented 6 years ago

I tested deploy.prototxt on Caffe, which is no problem.I have added IPlugin to tensorrt's deploy.prototxt.