The combination of Computer-Generated Holography (CGH) and deep learning has opened the possibility to generate both real-time and high-quality holograms. However, the widely-used data-driven deep learning method faces the problem of the large number of labeled training datasets generated by traditional algorithms, such as Gerchberg–Saxton (GS) iterative algorithm. It always takes a long time and limits the training performance of the network. In this work, we propose a model-driven neural network for high-fidelity Phase-Only Hologram (POH) generation. The Fresnel diffraction process is introduced as the physical model, which makes the network can automatically learn the latent encodings of POHs in an unsupervised way. Furthermore, the sub-pixel convolution upsampling method effectively improves the reconstruction quality. Once the training is completed, the POH of any two-dimensional image can be quickly generated. The calculation time is one to two orders of magnitude faster.
|