Skip to content

Latest commit

 

History

History
27 lines (19 loc) · 1.09 KB

README.md

File metadata and controls

27 lines (19 loc) · 1.09 KB

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:

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

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

Installation

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