search

chenzhi1992 / TensorRT-SSD

Use TensorRT API to implement Caffe-SSD， SSD（channel pruning）， Mobilenet-SSD
251 stars 84 forks source link

Error in Detectionout layer #28

Open ManishSoni1908 opened 6 years ago

ManishSoni1908 commented 6 years ago

@chenzhi1992 Thanks for the implementation of SSD Tensorrt. But i am getting the error when i trained the network on custom dataset with 6 classes, detection out layer return only zero values.

I have seen mbox_loc and mbox_conf output, they are non zero.

Thanks