Awesome-RWKV-in-Vision

RWKV4:

RWKV5&6:

RWKV7:

RWKV Ecosystem: https://rwkv.cn/eco

Table of Contents

• Vision Backbone
• Image Restoration
• Image Compression
• Image Generation
• Video Generation
• Image Segmentation
• Vision-Language Model
• 3D Point Cloud Learning
• Video Understanding
• Survey

Vision Backbone

Title	Code	Link
Vision-RWKV: Efficient and Scalable Visual Perception with RWKV-Like Architectures

Image Restoration

Title	Code	Link
Restore-RWKV: Efficient and Effective Medical Image Restoration with RWKV
GDSR: Global-Detail Integration through Dual-Branch Network with Wavelet Losses for Remote Sensing Image Super-Resolution	N/A
Exploring Linear Attention Alternative for Single Image Super-Resolution
URWKV: Unified RWKV Model with Multi-state Perspective for Low-light Image Restoration
Multi-View Learning with Context-Guided Receptance for Image Denoising

Image Compression

Title	Code	Link
Linear Attention Modeling for Learned Image Compression

Image Generation

Title	Code	Link
Diffusion-RWKV: Scaling RWKV-Like Architectures for Diffusion Models

Video Generation

Title	Code	Link
FEAT: Full-Dimensional Efficient Attention Transformer for Medical Video Generation

Image Segmentation

Title	Code	Link
Vision-RWKV: Efficient and Scalable Visual Perception with RWKV-Like Architectures
Mamba or RWKV: Exploring High-Quality and High-Efficiency Segment Anything Model
BSBP-RWKV: Background Suppression with Boundary Preservation for Efficient Medical Image Segmentation	N/A	ACMMM
RWKV-UNet: Improving UNet with Long-Range Cooperation for Effective Medical Image Segmentation
Zig-RiR: Zigzag RWKV-in-RWKV for Efficient Medical Image Segmentation		IEEE-TMI
ScaleMatch: Multi-scale Consistency Enhancement for Semi-supervised Semantic Segmentation		AAAI
RSRWKV: A Linear-Complexity 2D Attention Mechanism for Efficient Remote Sensing Vision Task	N/A

Vision-Language Model

Title	Code	Link
VisualRWKV: Visual Language model based on RWKV
RWKV-CLIP: A Robust Vision-Language Representation Learner

3D Point Cloud Learning

Title	Code	Link
PointRWKV: Efficient RWKV-Like Model for Hierarchical Point Cloud Learning
LION: Linear Group RNN for 3D Object Detection in Point Clouds
OccRWKV: Rethinking Efficient 3D Semantic Occupancy Prediction with Linear Complexity

Video Understanding

Title	Code	Link
Video RWKV: Video Action Recognition Based RWKV	N/A

Survey

Title	Code	Link
A Survey of RWKV
From Transformers to the Future: An In-Depth Exploration of Modern Language Model Architectures	N/A	LINK
The Evolution of RWKV: Advancements in Efficient Language Modeling	N/A