SSD-Centerloss
SSD-Centerloss copied to clipboard
question about low mAP and test.prototxt
Thanks for your work. I added your code to my SSD and trained on my own data, but I found some problems.
-
After 50000 iters, the mbox_loss decreased to 5~8, but mAP was always around 0.1 and did not increase. I wonder whether you have encountered such problem and know possible cause?
-
I changed
train.prototxt
according to the steps, but I did not changetest.prototxt
because it seems that detection_out layer should also be rewritten so that it could handle the input from its previous layer. So is it right not changingtest.prototxt
? Or should I rewrite detection_out?
@NovaBurst I have uploaded the center_loss_train.prototxt file and center_loss_test.prototxt file. And you can configure your prototxt file according to my example files. And I think center_loss is used to offer the feature maps more category features in the training stage, so the center_features and corresponding conv layers can be removed in the test stage.