image-restoration-sde
image-restoration-sde copied to clipboard
使用新的加噪策略在测试时遇到尺寸不兼容问题
学长你好!你们的工作是我今年复现的图像恢复代码里最棒的! 我对你们的模型做了一些魔改,在给LQ加噪时采用了一种特殊的加噪策略,最新的测试结果显示在LOL数据集上指标有较小提升,不过是在裁剪的128×128尺寸上的。 具体是在sde_util文件的generate_random_states和noise_state函数里修改,在生成噪声时产生了与输入张量尺寸相匹配的其它形式的噪声。但这样只能适用于128×128的尺寸,我想请教下你们是怎么在其他尺寸上做到的测试。 谢谢!