Annotate Data for Object Detection

First Look

You will need to have a computer with Nvidia GPU, should install CUDA, CUDNN, and use Module Downloader Window.

The AugeLab Studio Image Annotation Window allows users to annotate images by drawing bounding boxes around objects of interest and associating them with specific classes.

Getting Started

To open the Image Annotation Window, navigate to the top menu and click on AI Tools ➡️ Image Annotation.

For image annotation, you need two things:

1. .class file

2. Dataset

Class File

To label your data, first you need a classes.names file, which is a standard text file with .names extension. A normal class file looks like below:

Human
Dog
Cat
Cup

If you do not have such file, you can create your own using the Classes section:

To create your own classes file:

1. Type a class name

2. Click on + and add your classes.

3. Click on Save Classes and you are ready to pick a folder.

You may also click on - to remove any unwanted classes.

**Load Image Folder**

Make sure path to your dataset do not contain any non-english characters.

Click on Open Folder at the top of the screen and choose the folder that contains all of your images:

Select Image from List: After loading the image folder, a list of available images will be displayed. Click on an image to select it for annotation.

**Annotating Images**

Annotating images are pretty simple. Click on the top left of the object you'd like to detect, drag the mouse and release it when you are done!

Using the Dataset Panel

Dataset features several functionalities:

1. Filter function that allows you to filter several image classes:

   • All All images with and without annotation

   • Annotated images have annotations with them.

   • Empty images do no contain annotations, but included in the training. This means objects that are not annotated may negatively impact training.

   • Excluded images do no have an annotation file. This means they do not affect training whatsoever.

2. Search functionality will allow you to filter images with their names.

Annotating Videos

You can also annotate video files using the Video mode on the top side of the window:

Changing video mode will ask you for a file path. Choosing the video will allow you to annotate a video just like a folder!

Tools

There are several tools inside the Annotation Tool to help you during your dataset preparation:

Class Frequency Analysis

Clicking on class frequency analysis will analyze and show you how many classes exist in your dataset.

This is useful to check if you have a balanced dataset or not.

Augment Dataset

AugeLab Studio automatically applies dataset augmentation. Augmentation is the process of artificially creating similar data.

Clicking on Apply Augmentation will create artificial data and keep your annotations on the newly created images.

Augmentation process should be done after finishing annotation

Augmentation process may increase the disk size of your dataset up to 10 times.

Preprocess Image

Preprocess Image tool allows you to change the contrast, brightness and gamma of images that are shown in the window. This feature comes in handy when dealing with very dark or too bright images.

Change Class Id

Change Class Id tool will allow you to change the all annotated class instances to a different class.

This tool comes in hand when merging two different datasets.

Shortcuts and Help

For shorcuts and help, you can click on the `Help` button at the top menu.

• D: Show next image or frame.

• A: Show previous image or frame.

• Shift + D: Move forward by 10 images/frames.

• Shift + A: Move backward by 10 images/frames.

• W: Decrement class selection.

• S: Increment class selection.

• Shift + W: Decrement class selection by 3.

• Shift + S: Increment class selection by 3.

• X: Remove the last bounding box annotation.

• Shift + C: Clear all annotations.

• O: Add an empty annotation file or clear annotations.

• P: Remove annotations and clear the file.

• M: Move or exclude image to another folder (Folder Mode only).

• Shift + Delete: Remove image and annotation from computer (Folder Mode only).

Training With Custom AI Object Detection Model

To train a custom object detection model, please refer to Object Detection Train.