← all repositories

ozan-oktay/Attention-Gated-Networks

PyTorch implementation of attention gates integrated into U-Net and VGG-16 for medical image classification and segmentation.

2.1k stars Python Computer VisionML Frameworks
Attention-Gated-Networks
Velocity · 7d
+0.7
★ / day
Trend
steady
star history

This repository provides a PyTorch framework implementing attention gates used in convolutional neural networks for medical imaging tasks. It integrates attention mechanisms into U-Net and VGG-16 architectures, enabling both classification and segmentation of medical images such as ultrasound scan planes. The implementation supports additive attention gate mechanisms to improve model focus on relevant image regions.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.