Image creation for computer vision model training

A device, comprising:

• one or more memories;
• and
• one or more processors, communicatively coupled to the one or more memories, configured to:
• receive, from a user device, a one or more input images of an object of interest;
• obtain one or more background images, wherein the one or more background images include images relevant to the object of interest,
• wherein the one or more processors, to obtain the one or more background images, are configured to:
• determine a category associated with the object of interest,
• determine a background category corresponding to images of environments or locations associated with where the category of the object of interest is typically found or located,
• compile a database that correlates the background category to the category associated with the object of interest, and
• search for the one or more background images based on determining the background category;
• perform, for an input image from the one or more input images, a pre-processing operation to generate a first image that isolates the object of interest from the input image;
• perform, for the input image, a processing operation to generate a training image, wherein the processing operation includes:
• modifying the first image to generate a modified first image,
• adding the modified first image to a background image from the one or more background images to generate a second image, and
• modifying the second image to generate the training image;
• generate a training set that includes one or more training images that are generated based on the one or more input images and the one or more background images;
• and
• provide the training set to a computer vision model to cause the computer vision model to be trained using the training set and to enable the computer vision model to identify the object of interest.

A non-transitory computer-readable medium storing a set of instructions, the set of instructions comprising:

• one or more instructions that, when executed by one or more processors of a device, cause the device to:
• receive, from a user device, a one or more input images of an object of interest;
• obtain one or more background images, wherein the one or more background images include images relevant to the object of interest,
• wherein the one or more instructions, that cause the one or more processors to obtain the one or more background images, cause the one or more processors to:
• determine a category associated with the object of interest,
• determine a background category corresponding to images of environments or locations associated with where the category of the object of interest is typically found or located,
• compile a database that correlates the background category to the category associated with the object of interest, and
• search for the one or more background images based on determining the background category;
• perform, for an input image from the one or more input images, a pre-processing operation to generate a first image that isolates the object of interest from the input image;
• perform, for the input image, a processing operation to generate a training image based on the first image and a background image from the one or more background images;
• generate a training set that includes one or more training images that are generated based on the one or more input images and the one or more background images;
• and
• provide the training set to a computer vision model to cause the computer vision model to be trained using the training set and to enable the computer vision model to identify the object of interest.