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

question about low mAP and test.prototxt

Open NovaBurst opened this issue 6 years ago • 1 comments

Thanks for your work. I added your code to my SSD and trained on my own data, but I found some problems.

  1. 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?

  2. I changed train.prototxt according to the steps, but I did not change test.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 changing test.prototxt ? Or should I rewrite detection_out?

NovaBurst avatar Jan 05 '19 12:01 NovaBurst

@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.

BOBrown avatar Jan 07 '19 09:01 BOBrown