Closed Feecuin closed 4 months ago
是的,不加RLP和RPIM两个模块的话,DM本身就是UNet和Uformer,直接训练即可。RLP需要迭代多次,所以训练起来会更慢,DM单独训会更快
老哥方便加个微信聊吗?我现在训的数据集发现咱这方法跑不过PReNet,有些细节想和你确认下我训没训错
如果老哥方便的话,我的微信号是ideaalne1
1k张图数量其实有点少了,可能并不能充分训练,尤其是Uformer做主干的时候。PReNet本身结构简单,性能也还行,训练数据量小的情况下可能反而会有优势
老哥,你这deraining module里的UNet和Uformer都是源论文一样的代码么,如果把你提出的两个块不加单独训的话是不是可以得到UNet和Uformer的数据,你对比方法贴的UNet和Uformer的数据是不是这样训的呀?我搞了个1k的数据集训你的方法,70秒出一轮,训Uformer40秒,UNet14秒...是不是哪里有问题,还是说就是这么快?