Object Detection
Object detection models are trained to identify and label specific objects in your data using bounding boxes. For example, an object detection model can be trained to detect apples, and then used within a video project to label instances of apples within videos.
Ontology shape | Supported? |
---|---|
Bounding box | ✅ |
Rotatable bounding box | ❌ |
Polygon | ✅ |
Polyline | ❌ |
Primitive | ❌ |
Keypoint | ❌ |
Bitmask | ✅ |
Framework and models
We support the PyTorch framework for object detection models.
Creating object detection models
To learn how to create object detection models, head over to our models page.
Working with object detection models
Once a model has been attached to a project it can be used to perform the functions it has been trained on. Inside the label editor, click the Automated labeling button highlighted in the image below.
Open the ‘Detection and segmentation’ section, as seen in the screenshot below.
-
Select the model you would like to run. You will be able to choose from a list of models previously attached to the project.
-
The ‘Detection range’ lets you determine the start and end frames you would like the model to run on.
-
Set the Confidence. A value ranging from 0 to 1 that represents how confident the model has to be in order for a particular data point to be included in its output. Read more about confidence values here.
Advanced settings
-
Set the Intersection over union threshold. This parameter specifies that any boxes or polygons with an amount of overlap higher than the specified threshold should be deleted.
-
Choose between GPU or CPU processing units. CPUs are designed to handle a wide-range of tasks quickly, but are limited in how many tasks can run at the same time. GPUs are designed to quickly render high-resolution images and video concurrently.
-
The Tracking enabled toggle determines whether objects are part of the same ‘instance’ or not. In other words, whether the model should attempt to tracking individual instances through frames, or create separate objects for each frame.
Was this page helpful?