search

ChenzhaoNju / SFGNet

10 stars 1 forks source link

感觉里面差点文件 #2

Open tanhao2509231391 opened 9 months ago

tanhao2509231391 commented 9 months ago

作者你好，感觉这个代码工程还是没有发完整啊，没有test.py文件，train.py提示没有SSIM

ChenzhaoNju commented 9 months ago

eval.py就是test的，train.py没有SSIM是什么意思，实时测试ssim和psnr用的是skimage包我看有呀

tanhao2509231391 commented 9 months ago

就是import那个地方有个大写的ssim

---原始邮件--- 发件人: "Chen @.> 发送时间: 2024年1月25日(周四) 晚上9:12 收件人: @.>; 抄送: @.**@.>; 主题: Re: [zhihefang/SFGNet] 感觉里面差点文件 (Issue #2)

eval.py就是test的，train.py没有SSIM是什么意思，实时测试ssim和psnr用的是skimage包我看有呀

— Reply to this email directly, view it on GitHub, or unsubscribe. You are receiving this because you authored the thread.Message ID: @.***>

ChenzhaoNju commented 9 months ago

删掉它就行，最后改了一下，忘了删了，不用它。

tanhao2509231391 commented 9 months ago

train.py最上方有个from SSIM import SSIM，这个SSIM应该是你自己写的py文件，不像torch里面自带的

tanhao2509231391 commented 9 months ago

如果删掉的话那个损失函数该用哪几个哎

ChenzhaoNju commented 9 months ago

对的对的，感觉直接调包更方便。损失函数不变，这个不影响的。

tanhao2509231391 commented 9 months ago

OK，那问题都解决了

tanhao2509231391 commented 9 months ago

还有还有，这个epochs真的需要1600吗，我看论文里面没提到啊

ChenzhaoNju commented 9 months ago

300个就够了，这个可以自己调的，100个基本上就能达到不错的结果了。1600是因为这个模型跑的很快，跑一个晚上正好能跑完。

tanhao2509231391 commented 9 months ago

好的好的

---原始邮件--- 发件人: "Chen @.> 发送时间: 2024年1月25日(周四) 晚上9:33 收件人: @.>; 抄送: @.**@.>; 主题: Re: [zhihefang/SFGNet] 感觉里面差点文件 (Issue #2)

300个就够了，这个可以自己调的，100个基本上就能达到不错的结果了。1600是因为这个模型跑的很快，跑一个晚上正好能跑完。

— Reply to this email directly, view it on GitHub, or unsubscribe. You are receiving this because you authored the thread.Message ID: @.***>