espcn的matlab代码RDN-TensorFlow
原始来源
CVPR
2018
论文的
TensorFlow
实现。
官方实施:
先决条件
Python-2.7(分支python3的Python-3.5)
TensorFlow-1.10.0
Numpy-1.14.5
OpenCV-2.4.9.1
h5py-2.6.0
用法
准备数据
从
下载
DIV2K
训练数据。
提取所有图像并将其放置在
RDN-TensorFlow/Train/DIV2K_train_HR
中。
火车
python
main.py
测试
python
main.py
--is_train=False
注意
如果要在生成训练数据和测试图像时使用
MATLAB
中的
resize
函数作为使用的预训练模型,则需要安装
,并使用选项--matlab_bicubic=True运行脚本。
如果你想直接把原始图像作为RDN的输入,你可以像python
main.py
--is_train=False
--is_eval=False
--test_img=Test/Set5/butterfly_GT.bmp这样运行脚本
2021-10-22 15:56:16
5.03MB
系统开源
1