Skip to content

Use of Attention Gates in a Convolutional Neural Network / Medical Image Classification and Segmentation

License

Notifications You must be signed in to change notification settings

yf817/Attention-Gated-Networks

 
 

Repository files navigation

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"
    https://openreview.net/pdf?id=BJtn7-3sM

  2. "Attention U-Net: Learning Where to Look for the Pancreas"
    https://openreview.net/pdf?id=Skft7cijM

Installation

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

About

Use of Attention Gates in a Convolutional Neural Network / Medical Image Classification and Segmentation

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 100.0%