Switch panel target identification and one-meter line segmentation method based on multi-task learning

A switch panel target identification and one-meter line segmentation method based on multi-task learning belongs to the technical field of artificial intelligence and building engineering, and comprises the following steps: step 1, establishing a switch panel and building one-meter line pixel level data set; 2, amplifying the image data set, and enhancing the generalization ability of the network; step 3, constructing a lightweight multi-task deep learning model combining building switch panel target identification and building one-meter line segmentation; 4, training on the designed deep learning model by using the image data set amplified in the step 2; and 5, performing target identification and image segmentation by using the trained model, and outputting a result. By establishing the deep learning model with high recognition rate, high segmentation accuracy, low network parameters and real-time performance, the recognition accuracy of the switch panel and the accuracy of one-meter line segmentation can b