Construction method and device of training set, equipment and storage medium
A technology for constructing methods and training sets, applied in the computer field, can solve the problems of difficulty in guaranteeing the accuracy of manual labeling, time-consuming, and high labor costs.
- Summary
- Abstract
- Description
- Claims
- Application Information
AI Technical Summary
Problems solved by technology
Method used
Image
Examples
Embodiment 1
[0052] figure 1 It is a flowchart of a method for constructing a training set in Embodiment 1 of the present application. This embodiment of the present application is applicable to the case of efficiently and accurately expanding the training set when the training set only includes a small number of marked images. The method passes The training set construction device executes, and the device is realized by software and / or hardware, and is specifically configured in an electronic device with a certain data computing capability.
[0053] Such as figure 1 A method for constructing a training set is shown, including:
[0054]S101. Acquire a training set, where the training set includes multiple labeled first images.
[0055] In this embodiment, the training set is used to train the classification model, and includes multiple images with labeled categories. The number of categories is at least two, such as cat category, dog category, and the like. For the convenience of descr...
Embodiment 2
[0069] figure 2 It is a flowchart of a method for constructing a training set in Embodiment 2 of the present application. The embodiment of the present application is optimized and improved on the basis of the technical solutions of the above-mentioned embodiments.
[0070] Further, the operation "marking the second image according to the category and image features of the second image" is refined into "extracting the image features of the second image; judging whether the image features of the second image meet the image feature conditions corresponding to the category ; If the image features of the second image do not meet the image feature conditions corresponding to the category, correct the category of the second image, and use the corrected category to label the second image, thereby obtaining an accurately labeled second image.
[0071] Further, after the operation "add the marked second image to the training set", add the operation "return to the operation of using th...
Embodiment 3
[0110] image 3 It is a structural diagram of a training set construction device in Embodiment 3 of the present application. The embodiment of the present application is applicable to the case of efficiently and accurately expanding the training set when the training set only includes a small number of marked images. The device uses Realized by software and / or hardware, and specifically configured in an electronic device with certain data computing capabilities.
[0111] Such as image 3 The shown construction device 300 of a training set includes: an acquisition module 301, a classification module 302, a labeling module 303 and an addition module 304; wherein,
[0112] An acquisition module 301, configured to acquire a training set, the training set includes a plurality of marked first images;
[0113] A classification module 302, configured to use the training set to train the classification model, and use the trained classification model to classify the unmarked second im...
PUM
Login to View More Abstract
Description
Claims
Application Information
Login to View More