Home

Awesome

Attention Gated Networks <br /> (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.

<p align="center"> <img src="figures/figure1.png" width="640"> <br /> <em> The schematics of the proposed Attention-Gated Sononet</em> </p> <p align="center"> <img src="figures/figure2.jpg" width="640"> <br /> <em> The schematics of the proposed additive attention gate</em> </p>

References:

  1. "Attention-Gated Networks for Improving Ultrasound Scan Plane Detection", MIDL'18, Amsterdam <br /> Conference Paper <br /> Conference Poster

  2. "Attention U-Net: Learning Where to Look for the Pancreas", MIDL'18, Amsterdam <br /> Conference Paper <br /> Conference Poster

Installation

pip install --process-dependency-links -e .