关于RWKV在视觉方面应用

鄙人正在设计一个基于RWKV的超分模型,对标MambaIR并参考了Vision-RWKV的模型结构,基于DF2K+RSSCN7数据集进行预训练跑分如下:

总体而言在图像超分方面超过Mamba,但仍有大量改进空间还请多多指教。代码由于甲方爸爸的要求暂时无法贴出来 :frowning:

不过鄙人在此向大家征求一下RWKV在视觉方面能参考的改进点。鄙人此举抛砖引玉,还请各位大佬提提建议

3 Likes

不知道这个 RWKV-IR 作品的思路是否对您的工作有帮助?

多谢,鄙人参考一下