Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Derainig module咨询 #9

Closed
Feecuin opened this issue Apr 22, 2024 · 4 comments
Closed

Derainig module咨询 #9

Feecuin opened this issue Apr 22, 2024 · 4 comments

Comments

@Feecuin
Copy link

Feecuin commented Apr 22, 2024

老哥，你这deraining module里的UNet和Uformer都是源论文一样的代码么，如果把你提出的两个块不加单独训的话是不是可以得到UNet和Uformer的数据，你对比方法贴的UNet和Uformer的数据是不是这样训的呀？我搞了个1k的数据集训你的方法，70秒出一轮，训Uformer40秒，UNet14秒...是不是哪里有问题，还是说就是这么快？
@zkawfanx
Copy link
Owner

是的，不加RLP和RPIM两个模块的话，DM本身就是UNet和Uformer，直接训练即可。RLP需要迭代多次，所以训练起来会更慢，DM单独训会更快

@Feecuin
Copy link
Author

Feecuin commented Apr 22, 2024

老哥方便加个微信聊吗？我现在训的数据集发现咱这方法跑不过PReNet，有些细节想和你确认下我训没训错

@Feecuin
Copy link
Author

Feecuin commented Apr 22, 2024

如果老哥方便的话，我的微信号是ideaalne1

@zkawfanx
Copy link
Owner

1k张图数量其实有点少了，可能并不能充分训练，尤其是Uformer做主干的时候。PReNet本身结构简单，性能也还行，训练数据量小的情况下可能反而会有优势

@zkawfanx zkawfanx closed this as completed May 8, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
2 participants
@Feecuin @zkawfanx