Neural network model for segmenting image and image segmentation method thereof

Abstract

The invention discloses a neural network model for segmenting an image, an image segmentation method and device thereof, and a readable storage medium, the neural network model comprises an intelligent selection module, and the intelligent selection module further comprises a feature extraction unit and an intelligent selection unit. As the feature extraction unit adopts multi-scale cavity convolution, information of different scales of an input feature map is obtained, and a large amount of rich feature information is provided for subsequent feature screening; and the intelligent selection unit carries out intelligent screening on the input feature map channel by training a weight value according to the size of the weight value, so that the intelligent selection module can reduce the parameter quantity and the calculated quantity while ensuring the segmentation precision. Therefore, the neural network model provided by the invention can quickly extract effective features of the imageby adopting the intelligent selection module, is small in calculation amount and few in model parameters, and is suitable for a mobile terminal.

Description

technical field [0001] The present application relates to the technical field of computer vision, in particular to a neural network model for image segmentation, image segmentation method, device and readable storage medium. Background technique [0002] At present, solving computer vision problems such as image classification, image segmentation and object detection through deep learning has become popular and achieved great success. [0003] Among them, image segmentation technology is an important research direction in the field of computer vision and an important part of image semantic understanding. Image segmentation refers to the process of dividing an image into several regions with similar properties. In recent years, image segmentation technology has developed by leaps and bounds. This technology-related scene object segmentation, human body front and background segmentation, human face parsing, 3D reconstruction, etc. The technology has been widely used in unmann...

AI Technical Summary

Problems solved by technology

[0006] The purpose of this application is to provide a neural network model, image segmentation method, device and readable storage medium for image segmentation, to solve the problem of large amount of calculation and large amount of model parameters in the current neural network model for image segmentation , the problem that it cannot run on mobile devices such as mobile phones

Images