search

zhoubolei / CAM

Class Activation Mapping
http://cnnlocalization.csail.mit.edu/
MIT License
1.84k stars 466 forks source link

Fine tuning #9

Closed mzahran001 closed 7 years ago

mzahran001 commented 7 years ago

How to fine tune your model? I don't have sufficient data to retrain your model from scratch.I want to fine tune your model on my data which has only two classes ?

zhoubolei commented 7 years ago

I included the train_val file and solver of the GoogleCAM. You just follow the standard fine-tuning procedure to finetune the googleCAM model.

If you are not familiar with fine-tuning in caffe, you could just google fine-tune in caffe, or take this page as example: http://caffe.berkeleyvision.org/gathered/examples/finetune_flickr_style.html