Skip to content

UNET implementation support #130

New issue

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

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

Already on GitHub? # to your account

Open
sidmishfpl1801 opened this issue May 17, 2023 · 2 comments
Open

UNET implementation support #130

sidmishfpl1801 opened this issue May 17, 2023 · 2 comments

Comments

@sidmishfpl1801
Copy link

sidmishfpl1801 commented May 17, 2023

Is it possible to utilize the pre-trained Keras UNET model for making predictions? If not, can we incorporate our own custom layer instead? If neither option is feasible, could you kindly provide some suggestions for a possible workaround?

@milancurcic
Copy link
Member

Currently no. I believe the key missing piece is the upsampling layer type, i.e. Keras's UpSampling2D layer. I'll open an issue for it.

@milancurcic
Copy link
Member

On second thought, a transpose convolution is also missing here. So we need both transpose operations of convolutions and pooling for a UNET to work.

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

No branches or pull requests

2 participants