Image Compressive Sensing With Adaptive Sampling by Median Filtering

Abstract

Deep unfolding compressive sensing (CS) has experienced remarkable advancements. However, there still exist two challenges: (1) Many algorithms either use uniform block-based sampling, which ignore the fact that the content of different blocks is different, or allocate the sampling rate referring to complete signal before CS sampling, which is not always feasible in real-world scenarios. (2) Traditional CNN is difficult to capture broader contextual priors during iterative recovery. In this paper, we propose a novel network ASMFNet to solve the above two issues. Specifically, to address the first issue, we introduce a dual-branch network featuring a basic sampling branch to acquire reference image and an adaptive sampling branch by median filtering for allocating remaining sampling rate adaptively. For the second problem, we use Swin Transformer and feature fusion block to increase the feature interactions. Experimental results demonstrate that our proposed method outperforms existing methods.