Joint Discontinuity-Aware Depth Map Super-Resolution Via Dual-Tasks Driven Unfolding Network

Lijun Zhao,Jialong Zhang,Jinjing Zhang,Huihui Bai,Anhong Wang
DOI: https://doi.org/10.1109/tim.2024.3417544
IF: 5.6
2024-01-01
IEEE Transactions on Instrumentation and Measurement
Abstract:Convolutional neural network (CNN) provides significant performance improvement for depth map super-resolution (DSR). However, CNN-based DSR methods still face two prominent issues: 1) the overall topology structure of CNN cannot be intuitively understood and 2) the boundary prior of depth maps is always not fully utilized or even ignored, which may make the reconstructed depth maps suffer from boundary blurring problem. In this article, we adopt dual-tasks alternating optimization approach to improve the depth discontinuity-preserving ability of CNN. Concretely, joint optimization problem of two specific tasks is split into two reliant suboptimization problems, which correspond to two suboptimization models: DSR suboptimization model and edge super-resolution (SR) suboptimization model. To achieve interpretability of the CNN, these two suboptimization models are unfolded into alternatively optimized DSR subnetwork and edge SR subnetwork (ENet) according to proximal gradient descent algorithm, which can jointly form edge-constrained DSR network (EC-DSRNet). The coarse-to-fine architecture of DSR subnetwork is designed to gradually capture structures from both color and edge features. Additionally, the proposed joint optimization model makes the proposed network explicitly solve two tasks in an iterative manner, which facilitates an intuitive understanding of the network. A large number of experimental results have demonstrated that the proposed EC-DSRNet can compete against many state-of-the-art DSR. Our codes are publicly released at https://github.com/mdcnn/EC-DSRNet.
What problem does this paper attempt to address?