YOLO | Object Detection

Overview

The YOLO | Object Detection plugin allows you to detect a variety of objects on image files in your project. The plugin will draw a bounding box around each object type you specify, using a bounding box tool of your choice.

Plugin Functionality

The YOLO Object Detection Plugin allows you to automatically detect objects in images or video frames using a pre-trained YOLO (You Only Look Once) model. It identifies and localizes objects by drawing bounding boxes around detected items and assigning them class labels with confidence scores.

Supported Data Types

• Image

• Video

Supported Annotation Tools

• Bounding Box

Plugin Configuration

The Overwrite setting in model plugins controls whether existing annotations are replaced or kept. When enabled, the plugin replaces all existing annotations with new model predictions; when disabled, it simply adds the new results without deleting what’s already there.

The Class Mapping setting defines how the model’s predicted classes are linked to your project’s label schema. Follow these steps to prepare your class mapping.

1. In the "Class Mapping" field, open the left dropdown, and pick from one of the classes the plugin can detect.

2. Open the right dropdown, and pick from one of the tools you have created in your project.

3. Click on the "plus" button to finalize the pairing. Now the object class and your bounding box tool are linked. The plugin will use the selected bounding box tool to label the selected category.

4. Starting again from Step 1, link as many tools to categories as needed.

YOLO | Object Detection plugin integration to the workflow

You may vary a number of settings related to your export from the Config JSON field. Each option is detailed below:

{
  "only_mapped_classes": false,
  "threshold": 0.5
}

• "only_mapped_classes": If this is set to true, the plugin will detect only the classes that are mapped in the class mapping. If it’s false, it will detect all supported classes, and any unmapped ones will be assigned to the "unknown" class.

  • Example:

    • "only_mapped_classes": true

    • "only_mapped_classes": false

• "threshold": The confidence score threshold for object detection. Detections with a confidence score below this value will be filtered out. A higher threshold reduces false positives but may miss low-confidence detections.

  • Example:

    • "threshold": 0.5

Detection Classes

The plugin can automatically detect the following object classes:

Traffic

person	bicycle	car
motorcycle	airplane	bus
truck	train	boat
traffic light	fire hydrant	stop sign
parking meter	bench

Animals

zebra	elephant	bird
cat	dog	horse
sheep	cow	giraffe

Clothes/Accessories

backpack	umbrella	handbag
tie	suitcase

Sports

frisbee	skis	snowboard
sports ball	kite	baseball bat
baseball glove	skateboard	surfboard
tennis racket

Kitchen

wine glass	cup	fork
knife	spoon	bowl

Food

banana	apple	sandwich
orange	broccoli	carrot
hot dog	pizza	donut
cake

Home

chair	couch	potted plant
bed	dining table	toilet
tv	microwave	oven
toaster	sink	refrigerator
book	clock	vase
scissors	teddy bear	hair drier
toothbrush

Electronics

laptop	mouse	remote
keyboard	cell phone