Collect images
To train an object detection model to recognize your objects, you have to gather images that contain those objects. Adhere to guidelines for image quantity and quality for better results.
Format and size
The images you'll feed your object detection model need these characteristics:
Format:
- JPG
- PNG
- BMP
Size:
- 6 MB maximum for training
- minimum width / height of 256 pixels x 256 pixels
Data quantity and data balance
It's important to upload enough images to train your AI model. A good starting point is to have at least 15 images per object for the training set. With fewer images, there's a strong risk that your model will learn concepts that are just noise, or not relevant. Training your model with more images should increase the accuracy.
Another consideration is to make sure your data is balanced. If you have 500 images for one object and only 50 images for another, your training dataset isn't balanced. This can cause the model to be better at recognizing one of the objects. For more consistent results, maintain at least a 1:2 ratio between the object with the fewest images versus the one with the most. For example, if the object with the greatest number of images has 500 images, the object with the fewest images should have at least 250 images for training.
Use more diverse images
Provide images that are representative of what will be submitted to the model during normal use. For example, let's say you're training a model to recognize apples. If you only train images of apples on plates, it might not consistently recognize apples in trees. Including different kinds of images will make sure that your model isn't biased and can generalize well. The following are some ways you can make your training set more diverse.
Background
Use images of your objects in front of different backgrounds—for example, fruit on plates, in hands, and on trees. Photos in context are better than photos in front of neutral backgrounds because they provide more information for the classifier.
Lighting
Use training images that have different lighting, especially if the images used for detection might have different lighting. For example, include images taken with flash, high exposure, and so on. It's also helpful to include images with varied saturation, hue, and brightness. Your device camera probably lets you control these settings.
Object size
Provide images in which the objects are of varied sizes, capturing different parts of the object—for example, a photo of bunches of bananas and a closeup of a single banana. Different sizing helps the model generalize better.
Camera angle
Try to provide images taken from different angles. If all your photos are from a set of fixed cameras such as surveillance cameras, assign a different label to each camera. This can help avoid modeling unrelated objects such as lampposts as the key feature. Assign camera labels even if the cameras capture the same objects.
Unexpected results
AI models can incorrectly learn characteristics that your images have in common. Let's say you want to create a model to distinguish apples from citrus. If you use images of apples in hands and of citrus on white plates, the model might train for hands versus white plates instead of apples versus citrus.
To correct this, use the above guidance on training with more varied images: provide images with different angles, backgrounds, object size, groups, and other variants.