├── README.md ├── data_loader.py ├── data_set └── Urben_pre │ ├── annotation │ ├── 0.png │ ├── 1.png │ ├── 10.png │ ├── 29.png │ └── 31.png │ ├── test │ ├── 29.png │ └── 31.png │ ├── train │ ├── 0.png │ └── 1.png │ └── valid │ ├── 1.png │ └── 10.png ├── dataset.py ├── enhance_image.py ├── evaluation.py ├── img ├── AttU-Net.png └── U-Net.png ├── main.py ├── models └── train.pth ├── network.py ├── result └── res.csv ├── solver.py ├── test ├── test_data.py ├── test_one_data.py └── test_train.py └── util └── data_preprocess.py /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QTazimi/Water_extraction/HEAD/README.md -------------------------------------------------------------------------------- /data_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QTazimi/Water_extraction/HEAD/data_loader.py -------------------------------------------------------------------------------- /data_set/Urben_pre/annotation/0.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QTazimi/Water_extraction/HEAD/data_set/Urben_pre/annotation/0.png -------------------------------------------------------------------------------- /data_set/Urben_pre/annotation/1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QTazimi/Water_extraction/HEAD/data_set/Urben_pre/annotation/1.png -------------------------------------------------------------------------------- /data_set/Urben_pre/annotation/10.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QTazimi/Water_extraction/HEAD/data_set/Urben_pre/annotation/10.png -------------------------------------------------------------------------------- /data_set/Urben_pre/annotation/29.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QTazimi/Water_extraction/HEAD/data_set/Urben_pre/annotation/29.png -------------------------------------------------------------------------------- /data_set/Urben_pre/annotation/31.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QTazimi/Water_extraction/HEAD/data_set/Urben_pre/annotation/31.png -------------------------------------------------------------------------------- /data_set/Urben_pre/test/29.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QTazimi/Water_extraction/HEAD/data_set/Urben_pre/test/29.png -------------------------------------------------------------------------------- /data_set/Urben_pre/test/31.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QTazimi/Water_extraction/HEAD/data_set/Urben_pre/test/31.png -------------------------------------------------------------------------------- /data_set/Urben_pre/train/0.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QTazimi/Water_extraction/HEAD/data_set/Urben_pre/train/0.png -------------------------------------------------------------------------------- /data_set/Urben_pre/train/1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QTazimi/Water_extraction/HEAD/data_set/Urben_pre/train/1.png -------------------------------------------------------------------------------- /data_set/Urben_pre/valid/1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QTazimi/Water_extraction/HEAD/data_set/Urben_pre/valid/1.png -------------------------------------------------------------------------------- /data_set/Urben_pre/valid/10.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QTazimi/Water_extraction/HEAD/data_set/Urben_pre/valid/10.png -------------------------------------------------------------------------------- /dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QTazimi/Water_extraction/HEAD/dataset.py -------------------------------------------------------------------------------- /enhance_image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QTazimi/Water_extraction/HEAD/enhance_image.py -------------------------------------------------------------------------------- /evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QTazimi/Water_extraction/HEAD/evaluation.py -------------------------------------------------------------------------------- /img/AttU-Net.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QTazimi/Water_extraction/HEAD/img/AttU-Net.png -------------------------------------------------------------------------------- /img/U-Net.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QTazimi/Water_extraction/HEAD/img/U-Net.png -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QTazimi/Water_extraction/HEAD/main.py -------------------------------------------------------------------------------- /models/train.pth: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /network.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QTazimi/Water_extraction/HEAD/network.py -------------------------------------------------------------------------------- /result/res.csv: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /solver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QTazimi/Water_extraction/HEAD/solver.py -------------------------------------------------------------------------------- /test/test_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QTazimi/Water_extraction/HEAD/test/test_data.py -------------------------------------------------------------------------------- /test/test_one_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QTazimi/Water_extraction/HEAD/test/test_one_data.py -------------------------------------------------------------------------------- /test/test_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QTazimi/Water_extraction/HEAD/test/test_train.py -------------------------------------------------------------------------------- /util/data_preprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/QTazimi/Water_extraction/HEAD/util/data_preprocess.py --------------------------------------------------------------------------------