search

sanghyun-son / EDSR-PyTorch

PyTorch version of the paper 'Enhanced Deep Residual Networks for Single Image Super-Resolution' (CVPRW 2017)
MIT License
2.44k stars 673 forks source link

怎么使用自定义的数据集训练？ #357

Open Liiiiaictx opened 9 months ago

Liiiiaictx commented 9 months ago

我更改了和DIV2K一样的结构，出现以下这种错误： (EDSR) root@autodl-container-6b4f11a552-34a21590:~/autodl-tmp/EDSR-PyTorch/src# python main.py --model EDSR --scale 2 --save edsr_x2 --n_resblocks 32 --n_feats 256 --res_scale 0.1 --reset Traceback (most recent call last): File "main.py", line 33, in main() File "main.py", line 22, in main loader = data.Data(args) File "/root/autodl-tmp/EDSR-PyTorch/src/data/init.py", line 23, in init m = import_module('data.' + module_name.lower()) File "/root/miniconda3/envs/EDSR/lib/python3.6/importlib/init.py", line 126, in import_module return _bootstrap._gcd_import(name[level:], package, level) File "", line 994, in _gcd_import File "", line 971, in _find_and_load File "", line 953, in _find_and_load_unlocked ModuleNotFoundError: No module named 'data.aid'

我的新数据集的目录结构如下： 22

Sherlock-hh commented 8 months ago

https://blog.csdn.net/qq_41890263/article/details/125790425

cjzhzh commented 5 months ago

我有个疑问啊 将epoch设置为300后 发现终端中只输出epoch1-299 究竟是运行了300epoch还是299epoch啊

Liiiiaictx commented 5 months ago

@cjzhzh 这个问题不大，你训练300轮用了多长时间啊

cjzhzh commented 5 months ago

我用的一篇基于EDSR的改进CARN 用的单卡4080 大概3个多小时

Liiiiaictx commented 5 months ago

@cjzhzh 你知道EDSR怎么添加SSIM的计算方法

cjzhzh commented 5 months ago

推理吗 还是训练？推理的话网上找一份算ssim的就行

Liiiiaictx commented 5 months ago

@cjzhzh EDSR训练的时候在验证阶段只计算了psnr,这个怎么添加SSIM计算。CARN有计算SSIM指标吗

cjzhzh commented 5 months ago

CARN在验证的时候也没有ssim的计算

cjzhzh commented 5 months ago

https://github.com/xindongzhang/ELAN 这个有

Liiiiaictx commented 5 months ago

@cjzhzh 好的感谢，兄弟加个好友讨论下呗qq:2814669702@qq.com