
RDN-pytorch:基于PyTorch的残差密集网络图像超分辨率实现(CVPR 2018)
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
RDN-pytorch是基于PyTorch框架实现的残差密集网络(Residual Dense Network)项目,用于提升图像超分辨率效果。该代码库实现了CVPR 2018论文中的方法。
RDN 存储库要求使用以下软件版本:PyTorch 1.0.0、脾气暴躁的1.15.4(这里的“脾气暴躁”可能是指另一个特定版本或组件,原文如此)、枕具5.4.1、h5py 2.8.0 和 tqdm 4.30.0。DIV2K和Set5数据集可以转换为HDF5格式,并且可以从指定链接下载相关文件。
以下是数据集的规模类型关联:
- DIV2K:训练用(x2, x3, x4)、评估用(x2, x3, x4)
- Set5:训练用、评估用
或者,您可以使用prepare.py脚本创建自定义数据集。运行命令如下所示:
```
python train.py --train-file BLAH_BLAH/DIV2K_x4.h5 \
--eval-file BLAH_BLAH/Set5_x4.h5 \
--outputs-dir BLAH_BLAH/outputs
```
全部评论 (0)
还没有任何评论哟~


