├── .gitignore ├── README.md ├── app_arguments.py ├── app_gui.py ├── asset ├── .gitkeep ├── GUI_01.png ├── GUI_02.png ├── GUI_03.png ├── GUI_04.png ├── epoch_005.png ├── epoch_100.png ├── epoch_200.png └── feature_image.png ├── data ├── .gitkeep ├── GT.png └── RGB.png ├── dataset.py ├── export_onnx.py ├── images ├── .gitkeep ├── GT.png ├── RGB.png └── image_data.md ├── loss.py ├── model.py ├── models ├── Adam.model.weights.pt └── model.py ├── notebook ├── Create_Aerial_Sample.ipynb ├── Create_RS_Sample.ipynb └── DataSets_Dataloaders.ipynb ├── output ├── .gitkeep ├── loss_plot.png ├── mask.png └── prediction.png ├── predict.py ├── requirements.txt ├── train.py ├── utils.py └── visualization.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cuicaihao/aerial-image-segmentation/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cuicaihao/aerial-image-segmentation/HEAD/README.md -------------------------------------------------------------------------------- /app_arguments.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cuicaihao/aerial-image-segmentation/HEAD/app_arguments.py -------------------------------------------------------------------------------- /app_gui.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cuicaihao/aerial-image-segmentation/HEAD/app_gui.py -------------------------------------------------------------------------------- /asset/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /asset/GUI_01.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cuicaihao/aerial-image-segmentation/HEAD/asset/GUI_01.png -------------------------------------------------------------------------------- /asset/GUI_02.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cuicaihao/aerial-image-segmentation/HEAD/asset/GUI_02.png -------------------------------------------------------------------------------- /asset/GUI_03.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cuicaihao/aerial-image-segmentation/HEAD/asset/GUI_03.png -------------------------------------------------------------------------------- /asset/GUI_04.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cuicaihao/aerial-image-segmentation/HEAD/asset/GUI_04.png -------------------------------------------------------------------------------- /asset/epoch_005.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cuicaihao/aerial-image-segmentation/HEAD/asset/epoch_005.png -------------------------------------------------------------------------------- /asset/epoch_100.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cuicaihao/aerial-image-segmentation/HEAD/asset/epoch_100.png -------------------------------------------------------------------------------- /asset/epoch_200.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cuicaihao/aerial-image-segmentation/HEAD/asset/epoch_200.png -------------------------------------------------------------------------------- /asset/feature_image.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cuicaihao/aerial-image-segmentation/HEAD/asset/feature_image.png -------------------------------------------------------------------------------- /data/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /data/GT.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cuicaihao/aerial-image-segmentation/HEAD/data/GT.png -------------------------------------------------------------------------------- /data/RGB.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cuicaihao/aerial-image-segmentation/HEAD/data/RGB.png -------------------------------------------------------------------------------- /dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cuicaihao/aerial-image-segmentation/HEAD/dataset.py -------------------------------------------------------------------------------- /export_onnx.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cuicaihao/aerial-image-segmentation/HEAD/export_onnx.py -------------------------------------------------------------------------------- /images/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /images/GT.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cuicaihao/aerial-image-segmentation/HEAD/images/GT.png -------------------------------------------------------------------------------- /images/RGB.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cuicaihao/aerial-image-segmentation/HEAD/images/RGB.png -------------------------------------------------------------------------------- /images/image_data.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cuicaihao/aerial-image-segmentation/HEAD/images/image_data.md -------------------------------------------------------------------------------- /loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cuicaihao/aerial-image-segmentation/HEAD/loss.py -------------------------------------------------------------------------------- /model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cuicaihao/aerial-image-segmentation/HEAD/model.py -------------------------------------------------------------------------------- /models/Adam.model.weights.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cuicaihao/aerial-image-segmentation/HEAD/models/Adam.model.weights.pt -------------------------------------------------------------------------------- /models/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cuicaihao/aerial-image-segmentation/HEAD/models/model.py -------------------------------------------------------------------------------- /notebook/Create_Aerial_Sample.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cuicaihao/aerial-image-segmentation/HEAD/notebook/Create_Aerial_Sample.ipynb -------------------------------------------------------------------------------- /notebook/Create_RS_Sample.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cuicaihao/aerial-image-segmentation/HEAD/notebook/Create_RS_Sample.ipynb -------------------------------------------------------------------------------- /notebook/DataSets_Dataloaders.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cuicaihao/aerial-image-segmentation/HEAD/notebook/DataSets_Dataloaders.ipynb -------------------------------------------------------------------------------- /output/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /output/loss_plot.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cuicaihao/aerial-image-segmentation/HEAD/output/loss_plot.png -------------------------------------------------------------------------------- /output/mask.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cuicaihao/aerial-image-segmentation/HEAD/output/mask.png -------------------------------------------------------------------------------- /output/prediction.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cuicaihao/aerial-image-segmentation/HEAD/output/prediction.png -------------------------------------------------------------------------------- /predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cuicaihao/aerial-image-segmentation/HEAD/predict.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cuicaihao/aerial-image-segmentation/HEAD/requirements.txt -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cuicaihao/aerial-image-segmentation/HEAD/train.py -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cuicaihao/aerial-image-segmentation/HEAD/utils.py -------------------------------------------------------------------------------- /visualization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cuicaihao/aerial-image-segmentation/HEAD/visualization.py --------------------------------------------------------------------------------