SA-UNet: Spatial Attention U-Net for Retinal Vessel Segmentation

This is an unofficial PyTorch implementation of SA-UNet.

Requirements

pip install -r requirements.txt

Model

Diagram of SA-UNet architecture.