Learned digital lens enabled single optics achromatic imaging

Tianyue He, Qican Zhang, Chongyang Zhang, TingdongKou, SHEN JUNFEI

Abstract: High-quality imaging with reduced optical complexity has been extensively investigated owing to its promising future in academic and industrial research. However, the practical performance of most imaging systems has encountered a bottleneck posed by optics rather than electronics. Here, we propose a digital lens (DL) to compensate for the chromatic aberration induced by physical optical elements, while the residual wavelength-independent degradation is tackled through a self-designed neural network. By transforming physical aberration correction to an algorithm-based computational imaging task, the proposed DL enables our framework to reduce optical complexity and achieve achromatic imaging in the analog domain. Real experiments have been conducted with an off-the-shelf single lens and recovered images show up to 14.62 dB higher peak signal-to-noise ratio (PSNR) than the original chromatic input. Furthermore, we run a comprehensive ablation study to highlight the contribution of embedding the proposed DL, which shows a 4.83 dB PSNR improvement compared with the methods without DL. Technically, the proposed method can be an alternative for future applications that require both simple optics and high-fidelity visualization.