Learning data generation device, learning data generation method, and learning data generation program

Abstract

This enables the generation of effective image data to improve detection accuracy without human intervention. [Solution] The image conversion unit 21 receives image data, which is training data 31, as input to the image conversion model 41 that converts image data, and uses this training data to train an object detection model that detects objects with the attributes of the target object from the image data. The image conversion model 41 then converts the image data to obtain a converted image. The training data addition unit 22 adds the converted image obtained by the image conversion unit 21 to the training data 31.

Claims

1. An image conversion unit receives image data, which is training data for an object detection model, as input to an image conversion model that converts image data, and obtains a converted image obtained by the image conversion model. A learning data addition unit adds the converted image acquired by the image conversion unit to the learning data. A learning data generation device equipped with the following features.

2. The image conversion unit instructs the image conversion model to maintain the state in which the image data includes the specified attribute, and converts the image data. The learning data generation device according to claim 1.

3. The image conversion unit instructs the image conversion model to convert the image data by modifying other parts of the image data while leaving the parts relating to the specified attributes unchanged. The learning data generation device according to claim 1.

4. The image conversion model includes a text generation model that generates a descriptive text for image data, a document conversion model that converts the descriptive text, and an image generation model that generates image data from the descriptive text. The image conversion unit inputs the image data to the text generation model to obtain a description of the image data generated by the text generation model, inputs the acquired description to the document conversion model to obtain a converted text converted by the document conversion model, inputs the converted text converted to the image generation model to obtain the image data generated by the image generation model as the converted image. The learning data generation device according to claim 1.

5. The learning data generation device according to claim 4, wherein the image conversion unit instructs the document conversion model to convert the explanatory text by changing other parts without changing the part relating to the specified attribute in the explanatory text.

6. The aforementioned learning data generation device further, A data determination unit inputs the converted image acquired by the image conversion unit to the object detection model and determines whether the confidence level of object detection by the object detection model is equal to or greater than an evaluation threshold. Equipped with, The training data addition unit adds the converted image to the training data when the data determination unit determines that the confidence level is less than the evaluation threshold. The learning data generation device according to claim 1.

7. The data determination unit, when the confidence level is less than the evaluation threshold, inputs the converted image and the specified attribute to an attribute determination model that determines whether or not an object with a specific attribute exists in the image data, and obtains a determination result of whether or not an object with the specified attribute exists in the converted image. The training data addition unit adds the converted image to the training data when the data determination unit obtains a determination result that an object with the specified attribute exists in the converted image. The learning data generation device according to claim 6.

8. A computer inputs image data, which is training data to be used to train an object detection model, into an image transformation model that transforms image data, and obtains a transformed image obtained by the image transformation model. A method for generating training data in which a computer adds the converted image to the training data.

9. Image transformation processing involves inputting image data, which is training data for training an object detection model, to an image transformation model that transforms image data, and obtaining a transformed image obtained by the image transformation model. A training data addition process is performed to add the converted image obtained by the image conversion process to the training data. A training data generation program that makes a computer function as a training data generation device.

Images