README.md

ADGAN-Self-attention U-Net

Requirements

• Python 3.5
• PyTorch 1.0.0

pip install requirements.txt

Usage

Generate masks from trained models

1. https://drive.google.com/file/d/1UFyh-Sa2dsgyd8veDpqcvigdkmJcL1Xi/view?usp=sharing

run generate.py --experiment-name nodule2_64 --gpu

Output

Output will be stored in 'results/nodule2_64/test'