search

endernewton / tf-faster-rcnn

Tensorflow Faster RCNN for Object Detection
https://arxiv.org/pdf/1702.02138.pdf
MIT License
3.65k stars 1.57k forks source link

Fine tune Faster RCNN #322

Open satyakesav opened 6 years ago

satyakesav commented 6 years ago

Hi,

I want to fine tune the pre trained model on a new data set. Does the command mentioned for training "./experiments/scripts/train_faster_rcnn.sh 0 pascal_voc vgg16" actually tunes the model instead of training from scratch?

rnsandeep commented 6 years ago

If you have a pretrained model you can replace that in data/imagenet_weights, or it will finetune from the imagenet weights.

janevril commented 6 years ago

hi, do you know how to finetune all layers especially the conv3, conv4, conv5 before rpn? @rnsandeep @satyakesav ,it would be really helpful to me.

Fuhan1994 commented 5 years ago

hi, do you know how to finetune all layers especially the conv3, conv4, conv5 before rpn? @rnsandeep @satyakesav ,it would be really helpful to me.

have you solved this problem? How do you do?

Fuhan1994 commented 5 years ago

hello,i wonder if this code can be fine-tuned for the last few conv layers