I've trained SSD MobileNet v2 model using Tensorflow API on my own dataset of ~4k dog pictures and it displays bounding boxes all over the place. I've trained with batch size 1. enter image description here

The same dataset trained on faster rcnn works really well, and detects dogs properly. I've run 50k iterations, and total loss was spiking from 1.5 to 3 all the time in the end, with random bigger spikes, but overall it went down from 20

There wasn't a smooth and gradual decline, but a really spiky and uneven one with both models

Learning graph enter image description here

I know it's a fast model with low precision, but it should at least more or less detect the object. How can I increase accuracy so it at least partially detects objects without showing completely false bounding boxes ?

  • $\begingroup$ Hi there. You've described the problem adequately, now please add a question. – Reviewer $\endgroup$ – Jim Apr 30 '18 at 16:11
  • $\begingroup$ So you've got code that executes. The next step is to experiment with regularization and parameter tuning. $\endgroup$ – Sycorax Apr 30 '18 at 16:25

Your Answer

By clicking “Post Your Answer”, you agree to our terms of service, privacy policy and cookie policy

Browse other questions tagged or ask your own question.