daijifeng001 / R-FCN

R-FCN: Object Detection via Region-based Fully Convolutional Networks
Other
1.25k stars 543 forks source link

train on my own dataset #55

Open fbi0817 opened 7 years ago

fbi0817 commented 7 years ago

Hi ! @daijifeng001 Thank you for sharing this project for us! Now, I'd like to try training the model on my own dataset, and I notice there is a mean_image file. How to get that file if I want to train my own dataset, and your train process seems to use both voc2007 and voc2012, if I want to use only one dataset, which parts of code should I modify? Thank you!

liyi14 commented 7 years ago
  1. The mean_image file is calculated on the image-net dataset which is used to train the 'pre-trained network'. Since our weights are inherited from that network, we just keep it unchanged.
  2. Maybe you can remove one item in https://github.com/daijifeng001/R-FCN/blob/master/experiments/%2BDataset/voc0712_trainval_sp.m#L12
646677064 commented 7 years ago

@liyi14 Are the models whick the page provided trained using both voc2007 and voc2012? the models are different from the modes from kaiming He's page