Method for semantic segmentation based on knowledge distillation
Abstract:
A knowledge distillation based semantic image segmentation method includes inputting an input image to a teacher network and a student network; normalizing a first feature vector corresponding to each pixel in a feature map of a last layer of the teacher network and normalizing a second feature vector corresponding to each pixel in a feature map of a last layer of the student network; generating the first channel and space association matrix and the second channel and space association matrix based on the normalized first feature vector and the normalized second feature vector, and defining a first loss function based on a Euclidean norm value of the difference between the first channel and space association matrix and the second channel and space association matrix.
Public/Granted literature
Information query
Patent Agency Ranking
0/0