Train Custom AI Models

First Look

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

The AugeLab Studio Detection Training window provides a user-friendly interface to train object detection models using the YOLO (You Only Look Once) framework.

**Getting Started**

1. Launch AugeLab Studio on your computer.

2. Clic on AI Tools ➡️ Object Detection Training Widow
3. Before you start training, you need to load the dataset folder and the class file that contains the classes for the objects you want to detect.

You can integrate pretrained YOLO weights with class names as well.

**Menu Bar - Train Configuration**

1. Load Dataset: Click on this action to select the dataset folder containing images for training.

1. Load Class File: Use this action to load the class file (.names format) that contains the list of classes for training.

1. Detection Model Type: Choose between "Fast Detection" and "Robust Detection" models. The "Fast Detection" model is suitable for low-spec PCs and offers lower accuracy. The "Robust Detection" model provides higher accuracy but requires high-spec PCs.

1. Load Custom Training Checkpoint: If you have a pre-trained weight file (.weights or .pw format), you can load it to use as a starting point for training.

1. Open Advanced Settings: Further (advanced) configuration can be set in Advanced Settings window.

Advanced Settings Window

In this window, training parameters can be adjusted for advanced usage.

1. Dataset Split: Determines the proportion of data allocated for training and validation to evaluate the model's performance.

2. Network Size: Specifies the network size, impacting the trade-off between speed and accuracy in object detection tasks.

3. Subdivision Size: Defines the size of mini-batches during training, influencing memory usage and computational efficiency.

The "Train Procedure" action under the "Help" menu provides detailed instructions on how to train the models effectively.

**Starting Training Process**

1. Start Train: After loading the dataset and class file, click on START TRAIN to initiate the training process.

2. Train Log: The log area will display the progress and status of the training process.

3. Stop Train: You can stop the training process by clicking on the STOP TRAIN button.

You can follow training process from two screens:

1. Training Logging Window

Training Logging

1. Training Chart (Loss and mAP)

If you close the Detection Training window while the training process is running, the process will be terminated.

You can now use the Detection Training window in AugeLab Studio to train object detection models on custom datasets. Experiment with different model types, batch sizes, and custom checkpoints to achieve optimal results! Enjoy training your object detection models with AugeLab Studio!