论文标题
内容感知可扩展的深度压缩感应
Content-aware Scalable Deep Compressed Sensing
论文作者
论文摘要
为了更有效地解决图像压缩传感(CS)问题,我们提出了一种新颖的内容可扩展网络,该网络称为CASNET,该网络共同实现了自适应采样率分配,良好的粒度可伸缩性和高质量的重建。我们首先采用数据驱动的显着性检测器来评估不同图像区域的重要性,并提出基于显着性的块比率汇总(BRA)策略来分配采样率。然后开发一个统一的可学习生成矩阵,以产生具有有序结构的任何CS比的采样矩阵。 CASNet配备了以显着性信息和防止伪影的多块训练方案为指导的优化启发的恢复子网,CASNET共同重建了以一种模型以各种采样率采样的图像阻止图像。为了加速训练收敛并改善网络鲁棒性,我们提出了一种基于SVD的初始化方案和随机转换增强(RTE)策略,而无需引入额外的参数而无法扩展。所有CASNET组件都可以组合和端到端学习。我们进一步提供了四阶段的实施,用于评估和实际部署。实验表明,CASNET大量优于其他CS网络,从而验证了其组成部分和策略之间的协作和相互支持。代码可在https://github.com/guaishou74851/casnet上找到。
To more efficiently address image compressed sensing (CS) problems, we present a novel content-aware scalable network dubbed CASNet which collectively achieves adaptive sampling rate allocation, fine granular scalability and high-quality reconstruction. We first adopt a data-driven saliency detector to evaluate the importances of different image regions and propose a saliency-based block ratio aggregation (BRA) strategy for sampling rate allocation. A unified learnable generating matrix is then developed to produce sampling matrix of any CS ratio with an ordered structure. Being equipped with the optimization-inspired recovery subnet guided by saliency information and a multi-block training scheme preventing blocking artifacts, CASNet jointly reconstructs the image blocks sampled at various sampling rates with one single model. To accelerate training convergence and improve network robustness, we propose an SVD-based initialization scheme and a random transformation enhancement (RTE) strategy, which are extensible without introducing extra parameters. All the CASNet components can be combined and learned end-to-end. We further provide a four-stage implementation for evaluation and practical deployments. Experiments demonstrate that CASNet outperforms other CS networks by a large margin, validating the collaboration and mutual supports among its components and strategies. Codes are available at https://github.com/Guaishou74851/CASNet.