search

wkentaro / pytorch-fcn

PyTorch Implementation of Fully Convolutional Networks. (Training code to reproduce the original result is available.)
MIT License
1.74k stars 480 forks source link

Load FCN8 weights #145

Open luistelmocosta opened 4 years ago

luistelmocosta commented 4 years ago

How can I load FCN8 weights for a network with a different ouput than the one used to train original FCN8 on VOC dataset?

I downloaded the fcn8s_from_caffe.pth file and I want to load those weights on a network with num_classes=1.

How can I do that?

nothinghua commented 3 years ago

Can you send me the fcn8s_from_caffe.pth, my email is 1114605389@qq.com. If it is possible, I will be very grateful to you.