Dense-U-net: Dense Encoder–decoder Network for Holographic Imaging of 3D Particle Fields

Yufeng Wu,Jiachen Wu,Shangzhong Jin,Liangcai Cao,Guofan Jin

DOI: https://doi.org/10.1016/j.optcom.2021.126970

IF: 2.4

2021-01-01

Optics Communications

Abstract:Digital holographic imaging is able to reconstruct phase and three-dimensional (3D) information of an object from a one-shot two-dimensional (2D) lensless hologram. A dense encoder–decoder network, called Dense-U-net, is proposed to realize the reconstruction of a 3D particle field. The radii and 3D coordinates of the particles are encoded into a 2D grayscale image as the ground truth. The 2D hologram was served as the input of the network. A Dense_Block is designed and added to the encoder and decoder of the traditional U-net model. Each layer takes the inputs from all previous layers and passes its own feature map to all subsequent layers. Thus, as the diffraction is distributed throughout the entire hologram, both the features with high signal-to-noise ratio at the center of the particle and the diffraction features with low signal-to-noise ratio around the particle can be learned, thereby facilitating the full characterization of the particles. Using the specially designed dense connection module, network training can be accomplished with fewer parameters and faster speeds.