Train Your First Model
This section shows how to train and export an example deep learning model that can be used for defect segmentation. The data used for training is from an image dataset of Ethernet ports.
Preparation
Click here to download the image data and unzip the file.
Training Process
-
Create a New Project: Click New Project in the interface, name the project, and select a directory to save the project.
Do NOT include Chinese characters in the path name. -
Add the Defect Segmentation Module: Click
to add a module. Select Defect Segmentation and then click the OK button.
-
Import Image Data: Click Import in the upper left corner, select Folder and import the image dataset you have downloaded.
-
Image: Import image(s).
-
Folder: Import all images from the selected folder.
-
Dataset: A dataset contains images and corresponding labels. You can generate one by clicking .
-
-
Labeling: In this example, you will need to label the OK images and NG images in each dataset. OK means that the connectors meet quality requirements and NG means that there are defects such as deformations and fractures on the Ethernet ports. Labeling is to provide the information required by deep learning training.
For NG images, long press the left mouse button or right-click
on the toolbar and then use the Brush Tool to select the regions with defects.
Click
to use the eraser tool to remove the labeled region.For OK images that do not contain any defect, please select the image and then right-click and select Set to OK. Please make sure that there is at least one OK image in each dataset.
-
Train the Model: Select Training and then click Train in the lower right corner of the interface to start training the model.
-
Validate the Model: After the training is completed, click Validate to validate the model and check the results.
-
Export: Click Export in the lower right corner of the interface and click Export in the pop-up window (the default export parameters can be directly used). Then, select a directory to save the exported model in the dlkpack file.
Now, you have completed the training of the first model. Continue your reading to learn all the details of modules.