The dataset

The dataset contains 60 defective (NOK) and 40 non-defective (OK) rotor images. All images are annotated in LabelMe format; examples are shown below.

Load the annotated dataset

1. Request access to ReliVision Learning Center and download the rotor image set and its LabelMe annotation files.
2. In ReliUI, import the image set with Import Data.
3. Import the annotation file for each image folder with Import Annotated Dataset, choosing the LabelMe annotation format.
4. Your annotated dataset is now available in the Gallery.

Train the classification model

In ReliTrainer, train an AI block for rotor classification. The main steps are:

1. Select Classification as the model type.
2. Select your annotated rotor dataset.
3. Split the data into train and test sets, automatically or manually, to hold back some data for validation.
4. Set the hyperparameters and start training.

Hyperparameters used in this use case

• Epochs: 100 (default)
• X-Y resolution: 224
• Learning rate: 0.0001 (default)
• Momentum: 0.99 (default)
• Weight decay: 0.0005 (default)

Build and run the pipeline

Use the pipeline editor to build the pipeline by dragging and dropping AI and basic blocks. For this use case you need an input data source and a classification AI block connected in series.

1. Open the pipeline editor.
2. Add an input data source and point it at your raw rotor image set.
3. Add the classification AI block you trained and connect it after the input.
4. Run the pipeline with the execute button at the top.

Review the results

Review the results in the ReliUI data annotation interface. The pipeline saves its outputs as a separate set of annotations for each image it runs on, so you can compare predictions against your labels image by image.

Sample NOK result

Sample OK result