tf-image-segmentation icon indicating copy to clipboard operation
tf-image-segmentation copied to clipboard

Published 20 hours ago • verified publisher icon warmspringwinds

Can we fine tune the FCN_8s network on our own dataset?

Open robot010 opened this issue 8 years ago • 2 comments

Hi, I was wondering if we can utilize the fcn_8s_train.py file to fine-tuned the FCN8s on our own dataset. It seems that when I do this, there is an incompatible shape problem because my dataset only has 7 classes. Besides, the fcn_8s_train.py file is trained on FCN_16s, am I right? So it might be improper for me to use that script. Do I have to use the tf_recored script to convert my dataset first? Any suggestion is welcome. Thanks.

robot010 avatar Mar 29 '17 20:03 robot010

Did you manage to tune the network with your dataset ? I also want to do that but can't find the right way.

MrChristo59 avatar Apr 11 '17 07:04 MrChristo59

@MrChristo59 I would like to suggest you check on this site: https://github.com/MarvinTeichmann/tensorflow-fcn, which I found out how to fine-tune on my own dataset. I noticed that for FCN, there is no need to have the deconv layer pre-trained, as long as we wanted to change the number of output neurons, we will need to train the deconv layer from stratch.

robot010 avatar Apr 11 '17 14:04 robot010