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

unet binary segmentation example #756

Merged
merged 11 commits into from
Mar 5, 2023
Merged

Conversation

lkdci
Copy link
Contributor

@lkdci lkdci commented Mar 1, 2023

Binary segmentation training example of UNet model on the Supervisely person dataset.
Also added arch_params, for vanilla UNet implementation with bilinear upsampling.

@dagshub
Copy link

dagshub bot commented Mar 1, 2023

Copy link
Contributor

@Louis-Dupont Louis-Dupont left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I like the refacto, it looks clean!
Just check my comment on ConvBaseStage.

@lkdci
Copy link
Contributor Author

lkdci commented Mar 5, 2023

Thanks for the review @Louis-Dupont , added make_downsample_module and DownSampleMode enum as suggested.

@lkdci lkdci requested a review from Louis-Dupont March 5, 2023 07:29
Copy link
Contributor

@Louis-Dupont Louis-Dupont left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM

@lkdci lkdci merged commit 44d6040 into master Mar 5, 2023
@lkdci lkdci deleted the feature/ALG-242_unet_example branch March 5, 2023 14:44
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

Successfully merging this pull request may close these issues.

2 participants