ozan-oktay/Attention-Gated-Networks
PyTorch implementation of attention gates integrated into U-Net and VGG-16 for medical image classification and segmentation.

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.