Model training method and device, point cloud missing completion method and device, equipment and medium

Abstract

The invention discloses a model training method and a device, a point cloud missing completion method and a device, electronic equipment and a computer readable storage medium. The model training method comprises the steps of obtaining training missing point cloud data; inputting the training missing point cloud data into an initial model to obtain training repair point cloud data, and adjusting parameters of the initial model based on the training repair point cloud data and original point cloud data corresponding to the training missing point cloud data; if it is detected that the training completion condition is met, determining that the initial model is a point cloud completion model; wherein the initial model comprises a target reconstruction network and an initial generation network, the target reconstruction network comprises a target coding network, the target coding network uses training missing point cloud data to carry out comparative learning, the training missing point cloud data is input into the target coding network to obtain input features, the input features are input into the initial generation network to obtain missing point cloud data, the missing point cloud data is used for generating training repair point cloud data; and the accuracy of the processed point cloud data after completion processing is improved.

Description

technical field [0001] The present application relates to the field of computer technology, and in particular to a model training, point cloud missing completion method, device, electronic equipment, and computer-readable storage medium. Background technique [0002] 3D reconstruction technology reconstructs 3D objects in the virtual world, which is the basis for the realization of 3D vision technologies such as VR / AR (Virtual Reality, Virtual Reality / Augmented Reality, Augmented Reality). In recent years, with the development of sensors and deep learning, 3D point cloud has become the mainstream representation of 3D reconstruction results. However, due to mutual occlusion between objects, technical limitations of hardware equipment, etc., the 3D reconstruction results based on point clouds are missing holes or shape structures. At present, some research work has proposed a point-based completion method, which can directly process point cloud data and obtain point cloud fea...

AI Technical Summary

Problems solved by technology

However, related technologies extract features from the missing input point cloud and obtain the feature representation of the input point cloud, which can only repair data from the perspective of existing data, thereby reducing the accuracy of the complementary data generated by the model

[0003] Therefore, the problem of low data accuracy in related technologies is a technical problem that those skilled in the art need to solve

Images