search

chuanqi305 / MobileNet-SSD

Caffe implementation of Google MobileNet SSD detection network, with pretrained weights on VOC0712 and mAP=0.727.
MIT License
2.04k stars 1.18k forks source link

Fine-tune MobileNet-SSD #57

Open didulle opened 6 years ago

didulle commented 6 years ago

Hello, I would like to fine-tune MobileNet-SSD with my own dataset and own classes. There is an example in Caffe for fine tuning an image classification pb (http://caffe.berkeleyvision.org/gathered/examples/finetune_flickr_style.html) but it is not suitable for a multiobject detection. Is there any code/example/tutorial available?

balajib363 commented 6 years ago

@didulle If you found any resource for custom classes training share here Thanks,

Challovactor commented 6 years ago

I also fine-tune my own model with my own datasets, my iterations is 240000, fixed other parameters, but the final model also can detect some objects that classes in pre-trained model, such as person, It can detect very well, but it's not in my classes. Does anyone know the reason?