Skip to content

WangXueyang-uestc/ARConv

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

24 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

[CVPR 2025] Adaptive Rectangular Convolution for Remote Sensing Pansharpening

Abstract: Recent advancements in convolutional neural network (CNN)-based techniques for remote sensing pansharpening have markedly enhanced image quality. However, conventional convolutional modules in these methods have two critical drawbacks. First, the sampling positions in convolution operations are confined to a fixed square window. Second, the number of sampling points is preset and remains unchanged. Given the diverse object sizes in remote sensing images, these rigid parameters lead to suboptimal feature extraction. To overcome these limitations, we introduce an innovative convolutional module, Adaptive Rectangular Convolution (ARConv). ARConv adaptively learns both the height and width of the convolutional kernel and dynamically adjusts the number of sampling points based on the learned scale. This approach enables ARConv to effectively capture scale-specific features of various objects within an image, optimizing kernel sizes and sampling locations. Additionally, we propose ARNet, a network architecture in which ARConv is the primary convolutional module. Extensive evaluations across multiple datasets reveal the superiority of our method in enhancing pansharpening performance over previous techniques. Ablation studies and visualization further confirm the efficacy of ARConv. The source is available at github.

🛠 Getting started

Setup environment

  1. clone the repository
git clone git@github.com:WangXueyang-uestc/ARConv.git
cd ARConv
  1. install dependencies
pip install -r requirements.txt

Prepare dataset

Datasets can be downloaded from the repo liangjiandeng/PanCollection. Remember to replace PATH TO TRAIN DATASET in .scripts with the path to the dataset.

🚀 Train the model

bash .scripts/train_{Datasets}.sh

👍 Additional Information

Weights:

Metrics:

  • MetricCode can be found here.
  • You can alse use the tools from liangjiandeng/DLPan-Toolbox (specifically, the 02-Test-toolbox-for-traditional-and-DL(Matlab) directory).

Parameter Count:

  • Since we have pre-initialized all the convolutional kernels in the model, directly computing the parameter quantity would be huge and unreasonable. This is because only a small part of them is involved in each step of update. Therefore, if you want to know about the parameter quantity of the model, please calculate the parameters of the used convolutional kernels after fixing the number of convolutional kernel sampling points.

📧 Contact

If you have any questions, please email xywang_uestc@std.uestc.edu.cn.

About

Official repo for Adaptive Rectangular Convolution

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Contributors 2

  •  
  •