EDSR MDSR IRAN RCN -pytorch实现及代码常见问题

2024-05-13 08:58:28

EDSR MDSR IRAN RCN -pytorch实现及代码常见问题

代码下载地址：https://github.com/sanghyun-son/EDSR-PyTorch

环境配置

Dependencies

Python 3.6
PyTorch >= 1.0.0
numpy
skimage
imageio
matplotlib
tqdm

配置代码

基于cuda=10.0安装pytorch 1.1.0

conda create -n edsr python=3.6.9conda activate edsrconda install pytorch==1.1.0 torchvision==0.3.0 cudatoolkit=10.0 -c pytorchpip install matplotlibpip install skimagepip install tqdm

ERROR: Could not find a version that satisfies the requirement skimage
ERROR: No matching distribution found for skimage
pip install scikit-image

训练

在下载的文件夹中打开demo.sh有相应的运行命令
如果只需要简单开始训练只需要 Python main.py
需下载相应的训练集，DIV2K：https://cv.snu.ac.kr/research/EDSR/DIV2K.tar
下载后将文件解压到对应文件位置EDSR-PyTorch-master\src\data，然后在option.py中更改数据路径。
训练自己的模型：Python main.py --scale 2 --model IRAN --n_resblocks 32 --n_feats 256 --res_scale 0.1

测试

下载有benchmark的数据集set5 set14 等：https://cv.snu.ac.kr/research/EDSR/benchmark.tar
在data目录下新建文件夹benchmark里面放入各个数据集即可，
配置参数：–scale 4 --pre_train …/experiment/test/model/model_best.pt --test_only --chop --data_test Set5
上面是已经训练过的模型结果model——best.
–scale 2 --pre_train …/experiment/test/model/EDSR_x2.pt --test_only --chop --data_test Set5 --model EDSR --n_resblocks 32 --n_feats 256 --res_scale 0.1
训练自己的模型

EDSR MDSR IRAN RCN -pytorch实现及代码常见问题相关推荐

pytorch lstm crf 代码理解
好久没有写博客了,这一次就将最近看的pytorch 教程中的lstm+crf的一些心得与困惑记录下来. 原文 PyTorch Tutorials 参考了很多其他大神的博客,https://blog.c ...
Resnet的pytorch官方实现代码解读
Resnet的pytorch官方实现代码解读目录 Resnet的pytorch官方实现代码解读前言概述 34层网络结构的"平原"网络与"残差"网络的结构图 ...
pytorch geometric GraphSAGE代码样例reddit和ogbn_products_sage，为何subgraph_loader将sizes设成[-1]
pytorch geometric GraphSAGE代码样例reddit和ogbn_products_sage,为何subgraph_loader将sizes设成[-1] loader infere ...
史上最详细的Pytorch版yolov3代码中文注释详解（四）
史上最详细的Pytorch版yolov3代码中文注释详解(一):https://blog.csdn.net/qq_34199326/article/details/84072505 史上最详细的Pyt ...
一步步读懂Pytorch Chatbot Tutorial代码(四) - 为模型准备数据
文章目录自述有用的工具代码出处目录头大代码及说明 Prepare Data for Models 重点关注 indexesFromSentence zeroPadding binaryMa ...
一步步读懂Pytorch Chatbot Tutorial代码(二) - 数据处理
文章目录自述代码出处目录代码 Create formatted data file (为了方便理解,把代码的顺序略微改一下, 此章节略长.) 1. `loadLines` 将文件的每一行拆分为 ...
一步步读懂Pytorch Chatbot Tutorial代码(三) - 创建字典
文章目录自述有用的工具代码出处目录代码 Load and trim data 类 class _ _ init _ _ 初始化实例变量 for word in sentence.split( ...
pytorch 车型分类代码
pytorch 车型分类代码 (1) 可以跑通,代码简洁学习自用持续更新 1 数据 1.1 文件配置数据需要按照下面的方式组织,在数据文件夹中需要有train和val文件夹,然后每个文件夹内中, ...
pytorch车牌识别代码
我不太了解pytorch车牌识别代码,但我可以提供一些关于它的基本信息. PyTorch是一种被广泛用于图像分类.语音识别和自然语言处理的深度学习框架.它可以用来构建.训练和测试复杂的模型,其中包括车 ...

最新文章

热门文章