qfgaohao / pytorch-ssd

MobileNetV1, MobileNetV2, VGG based SSD/SSD-lite implementation in Pytorch 1.0 / Pytorch 0.4. Out-of-box support for retraining on Open Images dataset. ONNX and Caffe2 support. Experiment Ideas like CoordConv.
https://medium.com/@smallfishbigsea/understand-ssd-and-implement-your-own-caa3232cd6ad
MIT License
1.39k stars 529 forks source link

validation loss get saturated #160

Open amindehnavi opened 3 years ago

amindehnavi commented 3 years ago

Hi everybody I use VOC dataset to train MobileNet_v2. first I separate the VOC dataset and use the images that contain only person labels. then I start training the MobileNet_v2 with these images and their corresponding labels, but after about 100 epochs, validation loss get saturated and doesn't decrease (I'm sure that my dataset is correct!).

To make sure the network is working properly, use different optimizer like Adam and learning rate schedulers like MULTISTEPLR. ONECYCLELR, REDUCELRONPLATEAU and COSINEANNEALINGLR all from Torch implemented schedulers. also I tested different scenarios in freezing backbone and extra layers. unfortunately in the best case the accuracy of trained net reached to %68 for single class detection i.e. Person. while in repository is said that they have reached to %68 on 20 class dataset and I expected to reach about %80 in accuracy for person detection (furthermore I use pretrained weights that exist in this repository).

does anyone has an idea what should I do to increase the accuracy?

Thanks