pytorch-ssd icon indicating copy to clipboard operation
pytorch-ssd copied to clipboard

How do we train on custom datasets

Open tiru1930 opened this issue 4 years ago • 2 comments

Hi

I have facemask dataset, i want to train ssd-mobilnet v1, I was able to train but my prediction are very bad.

for instance below is MAp

Average Precision Per-class: face: 0.0 face_mask: 1.937778212873579e-08

Average Precision Across All Classes:9.688891064367895e-09

So What steps i need to take care when i want to train on custom datasets.

  1. Do I need to change the priors ( anchor boxes)
  2. Can I use pre-trained SSD-mobilenetv1 or shall I start to using basenet alone.
  3. What kind of fine-tuning I can do to ?

tiru1930 avatar May 25 '20 11:05 tiru1930

I want to know too.

thimabru1010 avatar Aug 05 '20 19:08 thimabru1010

Did you get a solution? I am also facing the same problem.

bijjuair avatar Oct 10 '20 11:10 bijjuair