
小波MATLAB代码-WMCNN-Pytorch:基于Pytorch的WMCNN(利用小波多尺度卷积神经网络进行航空图像超分辨率)再现
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
小波MATLAB代码-WMCNN-Pytorch是一个项目,它使用Python的深度学习框架PyTorch来重现和实现一种名为WMCNN的小波多尺度卷积神经网络。该模型特别设计用于处理航空影像的超分辨率问题。
使用小波多尺度卷积神经网络(WMCNN)的航空图像超分辨率代码在Matlab和Pytorch中有不同的实现方式,请引用相关文献。
下表展示了RSSCN7数据集上不同方法的峰值信噪比(PSNR)值比较:
| 方法 | 提升因子 | 草地 | 地面 | 河湖 | 森林 | 居民区 | 停车场 | 平均数 |
|---------|----------|-------|--------|-------|-------|---------|---------|--------|
| WMCNN_paper2x | 38.82 | 37.30 | 28.35 | 32.41 | 29.68 | 28.49 | 29.10 | 32.02 |
| WMCNN_pytorch2x| 38.98 | 37.38 | 28.28 | 32.31 | 29.71 | 28.33 | 30.00 | 32.14 |
使用说明
首先,下载RSSCN7数据集,并将其放置在“data/rsscn7”目录下。然后,可以采用以下两种方式生成hdf5数据集:
- 使用Matlab:通过文件夹“matlab_generate_data”中的代码“generate_train.m”,来创建所需的hdf5数据集。
- 或者使用Python:如果无法利用Matlab,则可以通过运行python脚本“data_generator.py”来自动生成所需的数据集。
全部评论 (0)


