alexgkendall / SegNet-Tutorial

Files for a tutorial to train SegNet for road scenes using the CamVid dataset
http://mi.eng.cam.ac.uk/projects/segnet/tutorial.html
847 stars 517 forks source link

Fine-tune Segnet Pascal with VGG architecture #110

Open bhadresh74 opened 6 years ago

bhadresh74 commented 6 years ago

I was wondering if I can fine-tune the VGG architecture (Segnet_train.prototxt in this tutorial) with pre-trained segnet_pascal.caffemodel. I gave it a try and my per class accuracy is gradually decreasing after each iteration. But, over all accuracy is increasing which is good. But I would prefer that my both class has overall good accuracy rather the one doing good and one doing bad. Any help would be appreciated.