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

[Liiiiaictx]

我更改了和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'

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

[Sherlock-hh]

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

[cjzhzh]

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

[Liiiiaictx]

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

[cjzhzh]

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

[Liiiiaictx]

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

[cjzhzh]

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

[Liiiiaictx]

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

[cjzhzh]

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

[cjzhzh]

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

[Liiiiaictx]

@cjzhzh 好的感谢，兄弟加个好友讨论下呗qq:[email protected]