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

Provide weights for segmentation network #4

Open
metzzo opened this issue May 12, 2022 · 2 comments
Open

Provide weights for segmentation network #4

metzzo opened this issue May 12, 2022 · 2 comments

Comments

@metzzo
Copy link
Contributor

metzzo commented May 12, 2022

Hello!

Thank you for the interesting challenge!

I tried to reproduce your baseline method, but the weights are not publicly available. Training the network manually could be done, but complicates things for some people. For example, some do not have easy access to sufficient GPUs to train the model.

Would it be possible to share the weights of the semantic segmentation network with us?

Best regards,
Robert

@dzambrano
Copy link
Collaborator

hello! Thanks for the interest in this challenge!

While we're discussing about the possibility to have the pre-trained weights for the baseline, I would suggest to reduce the input resolution of the network. You can do that by reducing the MULTIPLICATIVE_FACTOR as:
INPUT.MULTIPLICATIVE_FACTOR = 1

You can check the config/defaults.py for details. This will reduce the input resolution but will keep the aspect ratio and generally the field of view.

Moreover, the baseline uses a pre-trained deeplabv3_resnet101; but you could consider to start from a shallower network.

@dzambrano
Copy link
Collaborator

A first baseline has been provided in the latest release v0.1.0

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