CNIPA.AI
返回搜索
档案

Compression and training method and apparatus for defect detection model

发明专利审中
4浏览
10权利要求 · 4 独立
§ Ⅰ

卷宗概要

IPC 分类

G6N 3/9G6N 3/464G6T 3/40G6T 7/G6V 10/74G6V 10/82G6V 20/70

CPC 分类

G6N3/9G6N3/464G6T3/40G6T7/8G6V10/761G6V10/82G6V20/70G6T2207/20081G6T2207/20084G6T2207/30108

Disclosed in the present application are a compression and training method and apparatus for a defect detection model. The method comprises: obtaining, by means of segmentation labeling, a segmentation labeling factor matrix of each sample image; inputting each sample image into both a first defect detection model and a second defect detection model, and extracting first feature maps outputted by target convolutional layers in the first defect detection model and second feature maps outputted by corresponding target convolutional layers in the second defect detection model; and calculating, by using the segmentation labeling factor matrix, corrected distances between corresponding feature vectors of the first feature maps and the second feature maps, and calculating, as a first loss function, the sum of the corrected distances between all the feature vectors of the first feature maps and the second feature maps. The present embodiment can improve the accuracy of detecting tiny product appearance defects by means of a compressed defect detection model.

原文(中文)

Disclosed in the present application are a compression and training method and apparatus for a defect detection model. The method comprises: obtaining, by means of segmentation labeling, a segmentation labeling factor matrix of each sample image; inputting each sample image into both a first defect detection model and a second defect detection model, and extracting first feature maps outputted by target convolutional layers in the first defect detection model and second feature maps outputted by corresponding target convolutional layers in the second defect detection model; and calculating, by using the segmentation labeling factor matrix, corrected distances between corresponding feature vectors of the first feature maps and the second feature maps, and calculating, as a first loss function, the sum of the corrected distances between all the feature vectors of the first feature maps and the second feature maps. The present embodiment can improve the accuracy of detecting tiny product appearance defects by means of a compressed defect detection model.