AndersonYong/URetinex-Net

初始化分解网络的训练疑问

Opened this issue · 8 comments

请问训练初始化分解网络只需要一个重建loss和L1(光照图,max(r,g,b))吗?我试了一下重建图像会出现色偏,求解惑

不太清楚你说的色彩偏差是有多严重,轻微的偏差是正常的

视觉上可以看出有明显色彩偏差,总是感觉重建出来的图偏向灰度图,丢失了颜色

另外,提供的decom.py网络的输出好像没有看到限制在0-1之间,是否是在其他地方进行了限制呢

视觉上可以看出有明显色彩偏差,总是感觉重建出来的图偏向灰度图,丢失了颜色

经过unfolding模块后的图像是会有一些色彩损失,不过从我们测试的数据集来看,没有看到偏向灰度图,你用的是什么数据集呢

另外,提供的decom.py网络的输出好像没有看到限制在0-1之间,是否是在其他地方进行了限制呢

网络的输出没有做特别的限制

请问能开源或分享一份初始化部分的训练代码吗?我在自己数据集上结果确实不正常,可能是代码问题,多谢了

可能我刚刚误解了,你说的色彩偏差是初始化模块的输出就已经偏差了吗

是的,图片分解为R和S,然后R*S重建出来的图就有色彩偏差了