Unlock instant, AI-driven research and patent intelligence for your innovation.

Classification model training method and device

A classification model and training method technology, applied in the field of image processing, can solve the problem of low label accuracy, and achieve the effect of improving classification accuracy

Active Publication Date: 2019-06-14
XIAMEN MEITUZHIJIA TECH
View PDF14 Cites 12 Cited by
  • Summary
  • Abstract
  • Description
  • Claims
  • Application Information

AI Technical Summary

Problems solved by technology

Therefore, the existing multi-label classification methods have the problem of low output label accuracy

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 model training method and device
  • Classification model training method and device
  • Classification model training method and device

Examples

Experimental program
Comparison scheme
Effect test

Embodiment Construction

[0051] In order to make the purpose, technical solutions and advantages of the embodiments of the present application clearer, the technical solutions in the embodiments of the present application will be clearly and completely described below in conjunction with the drawings in the embodiments of the present application.

[0052] When classifying images, a model will be trained first. In the prior art, when the trained model classifies images, there are generally two classification methods, one of which is the single-label classification method , in this method, each image hits only one label, that is, each image is only classified into one category, therefore, this classification method cannot fully express the semantics of the image.

[0053]Another classification method in the prior art is multi-label classification. In this method, the same image can be associated with multiple labels, that is, each image can be divided into multiple categories. However, in a multi-label ...

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 discloses a classification model training method and device, and the method and the device improve the precision of an obtained label through separately calculating the binary cross entropy of each label. In detail, the method comprises the steps of firstly acquiring a plurality of first training samples, wherein each first training sample comprises a first training image and a firstpreset number of labels corresponding to the first training image, and the first preset number of labels comprise labels corresponding to upper classifications and lower classifications correspondingto image contents respectively; then, carrying out machine learning training according to the plurality of first training samples to obtain an initial classification model; after the initial classification model is obtained, obtaining the binary cross entropy of each label in the initial classification model to serve as a sub-error value of the label; and finally, obtaining a total error value ofthe initial classification model according to the sub-error value of each label, and adjusting the initial classification model according to the total error value to obtain a target classification model.

Description

technical field [0001] The present application relates to the technical field of image processing, in particular, to a classification model training method and device. Background technique [0002] In the prior art, when classifying images, there are two common classification methods. One of the classification methods is the single-label classification method. In this method, each image only hits one label, that is, each An image is only divided into one category, so this classification method cannot fully express the semantics of the image; another classification method is multi-label classification, in which the same image can be hit with multiple labels, that is, Say, each image can be divided into multiple categories. In the existing multi-label classification algorithm, if the meaning corresponding to one label among the multiple labels corresponding to the same image is the upper classification of the meaning corresponding to another or more labels, that is, there is ...

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/62
Inventor 王辰龙
Owner XIAMEN MEITUZHIJIA TECH