├── .idea ├── SA-UNet.iml ├── deployment.xml ├── misc.xml ├── modules.xml ├── remote-mappings.xml └── vcs.xml ├── CHASE └── download ├── DRIVE └── download ├── Dropblock.py ├── Eval_chase.py ├── Eval_drive.py ├── Model ├── CHASE │ └── SA_UNet.h5 └── DRIVE │ └── SA_UNet.h5 ├── README.md ├── SA-UNet.png ├── SA_UNet.py ├── Spatial_Attention.py ├── Train_chase.py ├── Train_drive.py ├── flip.py ├── keras_dataAug.py ├── requirements.txt └── util.py /.idea/SA-UNet.iml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/clguo/SA-UNet/HEAD/.idea/SA-UNet.iml -------------------------------------------------------------------------------- /.idea/deployment.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/clguo/SA-UNet/HEAD/.idea/deployment.xml -------------------------------------------------------------------------------- /.idea/misc.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/clguo/SA-UNet/HEAD/.idea/misc.xml -------------------------------------------------------------------------------- /.idea/modules.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/clguo/SA-UNet/HEAD/.idea/modules.xml -------------------------------------------------------------------------------- /.idea/remote-mappings.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/clguo/SA-UNet/HEAD/.idea/remote-mappings.xml -------------------------------------------------------------------------------- /.idea/vcs.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/clguo/SA-UNet/HEAD/.idea/vcs.xml -------------------------------------------------------------------------------- /CHASE/download: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/clguo/SA-UNet/HEAD/CHASE/download -------------------------------------------------------------------------------- /DRIVE/download: -------------------------------------------------------------------------------- 1 | https://drive.google.com/file/d/1t_UxlVWZXBtJQQNxW0vPdwrnqcdYdrRs?usp=sharing 2 | -------------------------------------------------------------------------------- /Dropblock.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/clguo/SA-UNet/HEAD/Dropblock.py -------------------------------------------------------------------------------- /Eval_chase.py: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /Eval_drive.py: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /Model/CHASE/SA_UNet.h5: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/clguo/SA-UNet/HEAD/Model/CHASE/SA_UNet.h5 -------------------------------------------------------------------------------- /Model/DRIVE/SA_UNet.h5: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/clguo/SA-UNet/HEAD/Model/DRIVE/SA_UNet.h5 -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/clguo/SA-UNet/HEAD/README.md -------------------------------------------------------------------------------- /SA-UNet.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/clguo/SA-UNet/HEAD/SA-UNet.png -------------------------------------------------------------------------------- /SA_UNet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/clguo/SA-UNet/HEAD/SA_UNet.py -------------------------------------------------------------------------------- /Spatial_Attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/clguo/SA-UNet/HEAD/Spatial_Attention.py -------------------------------------------------------------------------------- /Train_chase.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/clguo/SA-UNet/HEAD/Train_chase.py -------------------------------------------------------------------------------- /Train_drive.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/clguo/SA-UNet/HEAD/Train_drive.py -------------------------------------------------------------------------------- /flip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/clguo/SA-UNet/HEAD/flip.py -------------------------------------------------------------------------------- /keras_dataAug.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/clguo/SA-UNet/HEAD/keras_dataAug.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/clguo/SA-UNet/HEAD/requirements.txt -------------------------------------------------------------------------------- /util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/clguo/SA-UNet/HEAD/util.py --------------------------------------------------------------------------------