Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

fine tune on pretrained KittiSeg weights #178

Open
yanweiw opened this issue Jul 10, 2018 · 0 comments
Open

fine tune on pretrained KittiSeg weights #178

yanweiw opened this issue Jul 10, 2018 · 0 comments

Comments

@yanweiw
Copy link

yanweiw commented Jul 10, 2018

Hi Marvin, could you talk about why you opt for fine tuning on the VGG16 weights rather than your KittiSeg weights? I have a binary segmentation dataset that's quite similar to your road segmentation data but with only 100 images. I was hoping to leverage tv-continue to fine tune on the fcn8 weights you provided. Can you give me some hints on how to do that?

Going off that, I was also hoping to reset the weights in final convolutional layers of fcn8 for better tuning towards my data domain. My ground truth labels are black/white binary masks, and I was wondering if I could avoid overlaying them on my original images to compile with the mask format required to train with your existing code? Could you provide some details of how you would go about it? Thanks a lot.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant