鄙人正在设计一个基于RWKV的超分模型,对标MambaIR并参考了Vision-RWKV的模型结构,基于DF2K+RSSCN7数据集进行预训练跑分如下:
总体而言在图像超分方面超过Mamba,但仍有大量改进空间还请多多指教。代码由于甲方爸爸的要求暂时无法贴出来
不过鄙人在此向大家征求一下RWKV在视觉方面能参考的改进点。鄙人此举抛砖引玉,还请各位大佬提提建议
不知道这个 RWKV-IR 作品的思路是否对您的工作有帮助?
多谢,鄙人参考一下