Welcome to OStack Knowledge Sharing Community for programmer and developer-Open, Learning and Share
Welcome To Ask or Share your Answers For Others

Categories

0 votes
918 views
in Technique[技术] by (71.8m points)

tensorflow - SSD-mobilenet object detection algorithm not converging

Hi I am trying to train SSD -mobilenet in-order to detect 13 classes. I also trained a faster rcnn -resnet101 .

My training data images have resolution of 265 * 450 . (most of them) and each class had 400 images.

Then this weird thing happened faster rcnn converged faster with batch size of 1 .

But my SSD didn't . It's not converging at all . Here I will put the two graphs ,

Purple - Faster Rcnn | resnet101 Blue - SSD | mobilenet

enter image description here

I have seen some have trained this for few classes and obtained good results . If anyone have obtained good results please answer following .

You had only one class , so How fast it converged ? What do you think , the network is suffering from vanishing gradents problem or local minima problem ?

Editing - Here I will update the mAp for each category plot .

mAp score for each category when using ssd with mobilenet

See Question&Answers more detail:os

与恶龙缠斗过久,自身亦成为恶龙;凝视深渊过久,深渊将回以凝视…
Welcome To Ask or Share your Answers For Others

1 Answer

0 votes
by (71.8m points)
Waitting for answers

与恶龙缠斗过久,自身亦成为恶龙;凝视深渊过久,深渊将回以凝视…
Welcome to OStack Knowledge Sharing Community for programmer and developer-Open, Learning and Share
Click Here to Ask a Question

...