Attention_Unet_Pytorch

Implementation of attention unet in pytorch.

Usage

python -m Attention_Unet_Pytorch.train

Attention map :

Segmentation :

Name		Name	Last commit message	Last commit date
Latest commit History 11 Commits
Attention_Unet_Pytorch		Attention_Unet_Pytorch
imgs		imgs
plots		plots
LICENSE		LICENSE
README.md		README.md
environment.yml		environment.yml
install.sh		install.sh
pyproject.toml		pyproject.toml