AI Classification (Multi-Class) Tool
Function Introduction
The AI Classification (Multi-Class) Tool is a deep-learning-based visual classification tool supporting custom training for up to 8 classes. Through a wizard workflow, you can quickly complete image capture, labeling, training, and inference deployment. Typical scenarios include mixed-model recognition and appearance-defect grading for a single workpiece family.
-
Capture Images: Collect representative training data from real production conditions.
-
Set Target Region: Configure ROI for subsequent labeling and training.
-
Label Images: Assign class labels to target regions (up to 8 classes).
-
Train and Validate: Train model, validate performance, and perform additional training if needed.
Workflow
After opening the tool, click New at the upper-right of model list to create and enter a new model workflow.
Capture Images
-
Ensure input image port is connected.
-
One image is captured automatically when opening the tool. Click Capture Image to acquire more images.
|
Include typical variations in data collection:
|
Set Target Region
|
If |
-
Click Edit to open target-region settings.
-
Choose region mode:
| Mode | Description |
|---|---|
Entire Image as Target Region |
ROI covers entire image. Suitable when target occupies most of image. |
Custom Target Region |
Use rectangle or circle tool to draw one ROI accurately around target area. |
-
Optionally set mask region to exclude irrelevant interference.
-
Click Save and Use to apply settings.
|
If target and mask layers completely overlap, only the top layer is editable. |
Label Images
-
Click Edit to enter labeling.
-
Create classes as needed (up to 8 classes) and name them.
-
Select ROI and click corresponding class
Label Imagebutton. -
Capture new images and repeat labeling until each class has sufficient samples.
|
Label diverse data for every class. Avoid ambiguous samples. If production lighting/angle varies, enable augmentation options such as brightness and rotation adaptation. |
-
Click Save and Use after labeling.
Train and Validate Model
-
Train model with Train.
-
Validate with Validate.
| Parameter | Description |
|---|---|
Validation Result |
Displays predicted class or unknown class. |
Confidence |
Confidence of current prediction. |
Time Cost |
Inference time per run (ms). |
Confidence Threshold |
Minimum confidence for class decision. Below threshold is judged as unknown class. Default value: 0.5 |
Heatmap |
Visualizes model attention regions when enabled. Default value: Disabled. |
-
Additional training (optional): Use
Additional Trainingto add misclassified or missing cases, then retrain and revalidate. -
Save and apply model: Click Save and Use, then select model in
Model Nameparameter for inference usage.