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

Using on high channel image dataset #15

Open
torjusn opened this issue Dec 3, 2020 · 1 comment
Open

Using on high channel image dataset #15

torjusn opened this issue Dec 3, 2020 · 1 comment

Comments

@torjusn
Copy link

torjusn commented Dec 3, 2020

I want to try e.g. FCN8 on a custom dataset of high channel images(100ch+) but I'm not used to image segmentation with pretrained encoders. Are the models w/ corresponding encoders for one particular dataset since the encoders are pretrained or could it work on a different dataset with much higher amount of channels?

@IanTaehoonYoo
Copy link
Owner

IanTaehoonYoo commented Dec 5, 2020

Hi, snoofalus

Pretrained encoders were trained based on the ImageNet dataset. If your dataset and ImageNet has very different features in images, it is better to avoid pretrained encoders.
You can set like 'pretrained = false', the encoder's weights are initialized based on an optimizer without pretrained weights.

    model = all_models.model_from_name[model_name](n_classes, batch_size,
                                                   pretrained=false)                                              

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

2 participants