Open Access

ARTICLE

Improved Lightweight Deep Learning Algorithm in 3D Reconstruction

Tao Zhang^1,*, Yi Cao²

1 School of Mechanical Engineering, North China University of Water Conservancy and Hydroelectric Power, Zhengzhou, 450045, China
2 Department of Electrical and Computer Engineering, University of Windsor, Windsor, ON, N9B 3P4, Canada

* Corresponding Author: Tao Zhang. Email:

Computers, Materials & Continua 2022, 72(3), 5315-5325. https://doi.org/10.32604/cmc.2022.027083

Received 10 January 2022; Accepted 04 March 2022; Issue published 21 April 2022

Abstract

The three-dimensional (3D) reconstruction technology based on structured light has been widely used in the field of industrial measurement due to its many advantages. Aiming at the problems of high mismatch rate and poor real-time performance caused by factors such as system jitter and noise, a lightweight stripe image feature extraction algorithm based on You Only Look Once v4 (YOLOv4) network is proposed. First, Mobilenetv3 is used as the backbone network to effectively extract features, and then the Mish activation function and Complete Intersection over Union (CIoU) loss function are used to calculate the improved target frame regression loss, which effectively improves the accuracy and real-time performance of feature detection. Simulation experiment results show that the model size after the improved algorithm is only 52 MB, the mean average accuracy (mAP) of fringe image data reconstruction reaches 82.11%, and the 3D point cloud restoration rate reaches 90.1%. Compared with the existing model, it has obvious advantages and can satisfy the accuracy and real-time requirements of reconstruction tasks in resource-constrained equipment.

---

Keywords

---