AA-TransUNet github： 用于预测任务的注意力增强的TransUNet

来源

github地址

AA_TransUNet架构

数据集和预训练模型

如果你对本文中使用的数据集（降水图和云量数据集）感兴趣，请访问SmaAt-UNet了解更多细节。

对于预训练的AA_TransUNet模型，请联系我们：yimin.yang@student.maastrichtuniversity.nl 或 siamak.mehrkanoon@maastrichtuniversity.nl。

请将数据集放到"\dataset “目录下进行训练和测试，并将预训练好的模型放到”\results\Model_Saved "目录下以便将来加载。

使用

所需的依赖性：

pip install -r requirements.txt

训练降水图数据集：

python train_precipitation.py

训练云图数据集：

python train_cloud_cover.py

在训练或加载预训练的模型后，你可以通过以下方式评估模型的性能：

评估降水图数据集：

python evaluate_precipitation.py

评估云图数据集：

python evaluate_cloud_cover.py

为了便于培训，我们还提供了一个colab演示：地址

作者

• Yimin Yang: yimin.yang@student.maastrichtuniversity.nl Siamak
• Mehrkanoon: siamak.mehrkanoon@maastrichtuniversity.nl