dyl96 / MSASDNet

A Shadow Detection Algorithm Based on Multi-Scale Spatial Attention Mechanism for Aerial Remote Sensing Images (IEEE GRSL 2022)

A Shadow Detection Algorithm Based on Multi-Scale Spatial Attention Mechanism for Aerial Remote Sensing Images

该论文发表在GRSL2022，欢迎引用！

论文地址：https://ieeexplore.ieee.org/document/9502720

数据集下载地址： https://pan.baidu.com/s/19AtO2Y7r5wDZs0PQgA5eug (password: mhxt)

数据集论文地址：https://www.sciencedirect.com/science/article/pii/S0924271620302045

如果你想使用预测脚本，将models/model.pth换成训练好的模型参数，运行prdict_single.py即可,

我训练好的模型下载地址：https://pan.baidu.com/s/1K6RAr813NTfUJLD1ALdH5A?pwd=7xic 提取码：7xic

python prdict_single.py

或者使用命令行改变图片和模型参数

python prdict_single.py --img_path x.tif --model_path y.pth

如果你想自己训练，先下载数据集，放到当前文件夹，使用train.py即可