Attention Gated Networks
(Image Classification & Segmentation)
Pytorch implementation of attention gates used in U-Net and VGG-16 models. The framework can be utilised in both medical image classification and segmentation tasks.
The schematics of the proposed Attention-Gated Sononet
The schematics of the proposed additive attention gate
References:
-
"Attention-Gated Networks for Improving Ultrasound Scan Plane Detection", MIDL'18, Amsterdam
Conference Paper
Conference Poster -
"Attention U-Net: Learning Where to Look for the Pancreas", MIDL'18, Amsterdam
Conference Paper
Conference Poster
Installation
pip install --process-dependency-links -e .