3D plane detection and reconstruction method and device and storage medium

The invention discloses a 3D plane detection and reconstruction method and device and a storage medium, and the method comprises the steps: inputting a to-be-detected image into a Swin Transform backbone network, and outputting a feature map with five scales; inputting the feature map into a feature pyramid network, performing linear interpolation on the feature map to be S * S * C, and generating Kernel G through six convolution layers and a 3 * 3 * D convolution layer; predicting a Center-press score by using a Center-press module, and multiplying the Center-press score by the predicted confidence coefficient score to obtain a confidence coefficient value; and multiplying the Kernel G by a mask F and an edge E generated by the mask head network to obtain a corresponding mask and an edge, and finally combining the obtained mask, the edge and a plane normal vector parameter to obtain a reconstructed image. According to the method, the probability that a non-planar region is wrongly predicted into a plane is g