Classification detection method and device based on lightweight classification model

A classification model and classification detection technology, applied in neural learning methods, biological neural network models, character and pattern recognition, etc., can solve the problems of low product image detection efficiency and low production efficiency, and achieve small initialization width and light classification model Quantity, good detection effect

Pending Publication Date: 2021-11-12
BEIJING LUSTER LIGHTTECH +1
View PDF0 Cites 1 Cited by
  • Summary
  • Abstract
  • Description
  • Claims
  • Application Information

AI Technical Summary

Problems solved by technology

[0005] This application provides a classification detection method and device based on a lightweight classification model to solve the problem that the detection method using a deep learning model in the prior art has low detection efficiency for product images, resulting in low production efficiency of products

Method used

the structure of the environmentally friendly knitted fabric provided by the present invention; figure 2 Flow chart of the yarn wrapping machine for environmentally friendly knitted fabrics and storage devices; image 3 Is the parameter map of the yarn covering machine
View more

Image

Smart Image Click on the blue labels to locate them in the text.
Viewing Examples
Smart Image
  • Classification detection method and device based on lightweight classification model
  • Classification detection method and device based on lightweight classification model
  • Classification detection method and device based on lightweight classification model

Examples

Experimental program
Comparison scheme
Effect test

Embodiment 1

[0066] Such as figure 1 As shown, Embodiment 1 provides a classification detection method based on a lightweight classification model, including the following steps:

[0067] S101. Acquire a training sample image, and determine a category of the training sample image.

[0068] In step S101, the training sample images are in different industrial inspection fields. In order to ensure the accuracy of the product category classification of the lightweight classification model built according to the actual detection situation in each industrial inspection field, a sufficient number training set images. Input enough training set images, that is, training sample images, into the constructed lightweight classification model, so that the lightweight classification model can learn enough training set images; that is, the characteristics of the training sample images, and extract them according to the detection Each training set image, that is, the characteristics of the training sampl...

Embodiment 2

[0113] Corresponding to the foregoing embodiment 1 of a classification detection method based on a lightweight classification model, the present application also provides a classification detection device based on a lightweight classification model. The classification detection device includes:

[0114] Training module and classification detection module;

[0115] Wherein, the training module is used for:

[0116] Obtain a training sample image, and determine the category of the training sample image;

[0117] Classifying and marking the training sample images according to the categories to obtain sample marked images;

[0118] Inputting the sample marked image into a lightweight classification model for training to obtain a trained lightweight classification model;

[0119] The classification detection module is used for:

[0120] Inputting the sample image to be evaluated into the trained lightweight classification model, judging and outputting the category to which the ...

the structure of the environmentally friendly knitted fabric provided by the present invention; figure 2 Flow chart of the yarn wrapping machine for environmentally friendly knitted fabrics and storage devices; image 3 Is the parameter map of the yarn covering machine
Login to view more

PUM

No PUM Login to view more

Abstract

The invention provides a classification detection method and device based on a lightweight classification model. The method comprises the steps of obtaining a training sample image, and determining the category of the training sample image; performing classification marking on the training sample image according to the category to obtain a sample marking image; inputting the sample marking image into the lightweight classification model for training to obtain a trained lightweight classification model; and inputting a to-be-evaluated sample image into the trained lightweight classification model, and judging and outputting the category of the to-be-evaluated sample image. According to the lightweight classification model, the model basic units in the trunk adopt a simple double-branch structure, transmission of to-be-evaluated sample image data is reserved while the number of the model basic units is increased, the parameter space can be lightened by using the convolution layer with the number of output channels equal to the number of convolution groups, the initialization width of the model is small, the model is lighter, so that the speed of processing the to-be-evaluated sample image data by the model is higher, the detection effect is better, and the detection efficiency is higher.

Description

technical field [0001] The present application relates to the technical field of deep learning models, in particular to a classification detection method and device based on a lightweight classification model. Background technique [0002] In the field of industrial inspection, due to the variety of surface defects of products. When classifying and detecting product surface defect categories, traditional recognition algorithms can be used to extract features from product images, and then surface defects and their categories can be determined according to the results of feature extraction. However, it is difficult to extract all effective features in product images by using traditional recognition algorithms such as grayscale extraction algorithms, area extraction algorithms, and contour extraction algorithms, and the detection effect on product images is poor. [0003] Since the deep learning model can automatically learn product image features, the product image is detecte...

Claims

the structure of the environmentally friendly knitted fabric provided by the present invention; figure 2 Flow chart of the yarn wrapping machine for environmentally friendly knitted fabrics and storage devices; image 3 Is the parameter map of the yarn covering machine
Login to view more

Application Information

Patent Timeline
no application Login to view more
Patent Type & Authority Applications(China)
IPC IPC(8): G06K9/62G06N3/04G06N3/08
CPCG06N3/08G06N3/045G06F18/241G06F18/214
Inventor 张黎姚毅杨艺全煜鸣金刚彭斌
Owner BEIJING LUSTER LIGHTTECH
Who we serve
  • R&D Engineer
  • R&D Manager
  • IP Professional
Why Eureka
  • Industry Leading Data Capabilities
  • Powerful AI technology
  • Patent DNA Extraction
Social media
Try Eureka
PatSnap group products

Browse by: Latest US Patents, China's latest patents, Technical Efficacy Thesaurus, Application Domain, Technology Topic.

© 2024 PatSnap. All rights reserved.Legal|Privacy policy|Modern Slavery Act Transparency Statement|Sitemap