search

Project-MONAI / MONAI

AI Toolkit for Healthcare Imaging
https://monai.io/
Apache License 2.0
5.74k stars 1.05k forks source link

ATTENTION BASED UNET #1201

Closed massimilianoporzio closed 2 years ago

massimilianoporzio commented 3 years ago

It would be very good if the project could provide a UNET model with attention mechanism like the one described in https://github.com/ozan-oktay/Attention-Gated-Networks. I think it would improve the performance of segmentation models

Nic-Ma commented 3 years ago

Hi @massimilianoporzio ,

Thanks for your suggestion here. I marked for working group discussion, and if you are interested, welcome to submit a PR for it.

Thanks.