
SmaAt-UNet:PyTorch代码实现基于论文“SmaAt-UNet”。
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
SmaAt-UNet 论文代码,名为“SmaAt-UNet:使用小型体系结构进行降水临近预报”,所建议的SmaAt-UNet模型位于指定的模型文件夹中。 本文详细阐述了我们采用的模块化方法(PL),该模块显著简化了训练流程,并极大地提升了添加记录器和生成检查点的便捷性。 为了实现PL的功能,我们构建了一个模型,其父级结构继承自pl.LightningModule。 这种模型与包含PL功能的纯PyTorch SmaAt-UNet实现完全一致。 训练过程以分类任务(PascalVOC)为例进行了演示。 为了完成降水任务的训练,我们使用了相应的图像文件。 该降水数据集包含了2016年至2019年期间每隔5分钟拍摄的雷达降水图,最终产生了大约420,000张图像。 数据集的设计基础是雷达降水图数据。 如示例所示,原始图像已经通过裁剪操作进行了处理。 如果您对我们所使用的数据集有进一步的疑问或感兴趣,可以通过电子邮件发送至: 和siamak.mehrkanoon@maastrichtuniversity.nl
全部评论 (0)
还没有任何评论哟~


