Low-cost and simple optical system based on wavefront coding and deep learning

With the development of computational imaging, the integration of optical system design and digital algorithms has made more imaging tasks easier to perform. Wavefront coding (WFC) is a typical computational imaging technique that is used to address the constraints of optical aperture and depth of field. In this paper, we demonstrated a low-cost and simple optical system based on WFC and deep learning. We constructed an optimized encoding method for the phase plate under the framework of deep learning, which reduces the requirement for aberration correction in the full field of view. Optical coding was achieved with just a double-bonded lens and a simple cubic phase mask, and digital decoding used the deep residual UNet++ network framework. The final image obtained has good resolution, whereas the depth of field of the system expanded by a factor of 13, which is of great significance for the high-precision inspection and attaching of small parts of machine vision.