Skip to content

Rep-Mamba: Re-parameterization in Vision Mamba for Lightweight Remote Sensing Image Super-Resolution

Notifications You must be signed in to change notification settings

meigeni0929/Rep-Mamba

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

5 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Rep-Mamba: Re-parameterization in Vision Mamba for Lightweight Remote Sensing Image Super-Resolution

Abstract

Selective space model (Mamba) has recently demonstrated great potential in remote sensing image super-resolution (RSISR) tasks due to its capability for long-range dependency modeling with linear computational complexity. Despite these merits, existing Mamba architectures face two critical challenges in large-scale remote sensing scenarios: i) Neglecting the local semantic integrity due to the unfolding 1D sequential representations; ii) Facing the dilemma between effectiveness and efficiency. To address these issues, we propose Rep-Mamba, a lightweight progressive multi-scale feature fusion architecture based on the State Space Model (SSM) for RSISR. Specifically, we innovatively design a Cross-Scale State Propagation (CSSP) mechanism and construct a Lightweight Progressive Fusion Module (LPFM) to dynamically capture hierarchical spatial dependencies in remote sensing scenes while maintaining high computational efficiency. Moreover, to achieve synergistic optimization between local semantic structure preservation and global context modeling, we introduce differentiable Re-parameterization Convolution (RepConv), which significantly enhances reconstruction accuracy and visual quality without compromising computational efficiency. Extensive experiments across multiple benchmarks demonstrate that Rep-Mamba achieves a superior trade-off between accuracy and complexity, highlighting its effectiveness and scalability.

Network

image

🧩 Install

git clone https://github.com/meigeni0929/Rep-Mamba.git

Environment

  • CUDA 11.8
  • Python 3.8
  • PyTorch 2.0.0
  • Torchvision 0.15.1
  • basicsr 1.4.2

🎁 Dataset

Please download the following remote sensing benchmarks:

Data Type AID DOTA-v1.0 DIOR NWPU-RESISC45
Training Download None None None
Testing Download Download Download Download

🧩 Test

  • Step I. Use the structure below to prepare your dataset.

/xxxx/xxx/ (your data path)

/GT/ 
   /000.png  
   /···.png  
   /099.png  
/LR/ 
   /000.png  
   /···.png  
   /099.png  
  • Step II. Change the --data_dir to your data path.
  • Step III. Run the eval_4x.py
python eval_4x.py

Train

python train.py

Contact

If you have any questions or suggestions, feel free to contact me.
Email: 24s103263@stu.hit.edu.cn , ymr2200642844@163.com

About

Rep-Mamba: Re-parameterization in Vision Mamba for Lightweight Remote Sensing Image Super-Resolution

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published