Train Custom AI Models
First Look

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**
Launch AugeLab Studio on your computer.
Clic on
AI Tools
➡️ Object Detection Training WidowBefore 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.
**Menu Bar - Train Configuration**
Load Dataset: Click on this action to select the dataset folder containing images for training.

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

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.

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.

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.
Dataset Split: Determines the proportion of data allocated for training and validation to evaluate the model's performance.
Network Size: Specifies the network size, impacting the trade-off between speed and accuracy in object detection tasks.
Subdivision Size: Defines the size of mini-batches during training, influencing memory usage and computational efficiency.

**Starting Training Process**
Start Train: After loading the dataset and class file, click on
START TRAIN
to initiate the training process.Train Log: The log area will display the progress and status of the training process.
Stop Train: You can stop the training process by clicking on the
STOP TRAIN
button.
You can follow training process from two screens:
Training Logging Window

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!
Last updated
Was this helpful?