Small object semantic segmentation method combined with object detection

Abstract

The invention discloses a small target semantic segmentation method combined with target detection. Attention semantic segmentation network, training the network to get the whole semantic segmentationmodel; making small target detection dataset and small target semantic segmentation dataset; training the small target detection network based on YOLOv2 through the small target detection data set; asmall target semantic segmentation network is designed and trained by using the small target semantic segmentation data set to obtain the small target semantic segmentation model. In the testing phase, the test image is used as the input of the whole semantic segmentation model and the small target detection network, and the segmentation result and the small target boundary box of the whole imageare obtained, which is modified by the small target semantic segmentation model. The invention can greatly reduce the segmentation difficulty of the small target, thereby effectively improving the segmentation performance of the small target.

Description

technical field [0001] The invention belongs to the technical field of image processing, and in particular relates to a small target semantic segmentation method combined with target detection. Background technique [0002] Image semantic segmentation is one of the three major tasks of computer vision. Its goal is to classify each pixel in the image and obtain a semantic segmentation map of the image. From the perspective of traditional image segmentation, image semantic segmentation is to segment the image into multiple regions at the semantic level, and then assign appropriate category labels to each region. At present, semantic segmentation has a wide range of applications in autonomous driving, real-time road monitoring, automatic virtual fitting, and medical disease systems. Before the rise of deep learning, the main method of semantic segmentation was to use the conditional random field model to build a probability graph model. In recent years, due to the strong learn...

AI Technical Summary

Problems solved by technology

However, the number of pixels contained in the area where the small target is located is much less than that contained in other targets. When the pixels in the sm...

Images