Use the Object Detection Module

You are viewing an old version of the documentation. You can switch to the documentation of the latest version by clicking the top-right corner of the page.

Please click here to download an image dataset of rotors. In this section, we will use an Object Detection module and train a model to detect the positions of rotors in the image and output the quantity.

You can also use your own data. The usage process is overall the same, but the labeling part is different.

Create a new project and add the Object Detection module: Click New Project in the interface, name the project, and select a directory to save the project. Click in the upper right corner of the Modules panel and add the Object Detection module.
Import the image dataset of rotors: Unzip the downloaded dataset file. Click the Import button in the upper left corner, select Folder, and import the image data.
Select an ROI: Click the ROI Tool button and adjust the frame to select the bin containing rotors in the image as an ROI, and click Apply to save the settings. Setting the ROI can avoid interferences from the background and reduce processing time.
Split the dataset into the training set and validation set: By default, 80% of the images in the dataset will be split into the training set, and the rest 20% will be split into the validation set. You can click and drag the slider to adjust the proportion. Please make sure that both the training set and validation set include objects of all classes to be detected. If the default training set and validation set cannot meet this requirement, please right-click the name of the image and then click Switch to training set or Switch to validation set to adjust the set to which the image belong.
Create Labels: Create labels based on the type or feature of different objects. In this example, the labels are named after the rotors.
Label images: Make rectangular selections on the images to label all the rotors. Please select the rotors as precisely as possible and avoid including irrelevant regions. Inaccurate labeling will affect the training result of the model. Click here to view how to use labeling tools.
Train the model: Keep the default training parameter settings and click Train to start training the model. Click here to view the instructions on training parameter configuration.
Validate the model: After the training is completed, click Validate to validate the model and check the results. Click here to view the instructions on validation parameter configuration.
Export the model: Click Export, set parameters of the model to be exported in the pop-up window, and then click Export and select a directory to save the exported model.