TianZerL / Anime4KCPP

A high performance anime upscaler
GNU General Public License v3.0
1.79k stars 137 forks source link

关于ACNet的训练 #71

Open FrankyCao opened 3 years ago

FrankyCao commented 3 years ago

数据集用DIV2K_train_HR,分辨率在720P以上,用之前提供的训练代码来跑了几百个epoch,loss并没有很好的收敛,有以下几个疑问望解答 1、卷积层权重的初始化方式和正态分布方式使用默认的?? 2、我看训练方式是使用的灰度图的数据,但是测试是只用了图片的亮度Y,这个有影响么? 3、500个epoch下来,loss在0.025左右,测试输出图像较模糊,达不到理想效果,应该是我的训练方式错了,有什么办法提高吗?

TianZerL commented 3 years ago

1.使用pytorch默认就行。 2.ACNet只处理灰度通道,Y不是亮度而是灰度。 3.ACNet就是这样,和fsrcnn一样很难训练。可能还比较挑训练集。