Use the Text Detection Module
Taking an image dataset of identification numbers (download) as an example, this section will show you how to use the Text Detection module to detect the text area in an image. The Text Detection module can be followed by a Text Recognition module to output the recognized numbers, letters, or some special symbols.
You can also use your own data. The usage process is overall the same, but the labeling part is different. |
Workflow
-
Create a new project and add the Text Detection module: Click New Project after you opened the software, name the project, and select a directory to save the project. Then, click in the upper-right corner and add the Text Detection module.
-
Import the image data of rotors: Unzip the downloaded data file. Click the Import/Export button in the upper left corner, select Import Folder, and import the image data.
When you select Import Dataset, you can only import datasets in the DLKDB format (.dlkdb), which are datasets exported from Mech-DLK. -
Select an ROI: Click the ROI Tool button and adjust the frame to set an ROI that covers the text areas of all images. Then, click Apply on the top of the selection region to save the setting. Setting the ROI can avoid interferences from the background.
-
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 all kinds of text areas 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 belongs.
-
Label the images: Select an appropriate tool from the toolbar to label the images.
-
Train the model: Keep the default training parameter settings and click Train to start training the model.
-
Validate the Model: After the training is completed, click Validate to validate the model and check the results.
After you validate a model, you can import new image data to the current module and use the pre-labeling feature to perform auto-labeling based on this model. For more information, see Pre-labeling.
-
Export the model: Click Export. Then, set Input image size in the pop-up dialog box, click Export, and select a directory to save the exported model.
By default, the Input image size is consistent with that set in the Validation Parameters dialog box. In validation, if the inference speed and accuracy are up to standard, you can keep the default setting.
The exported model can be used in Mech-Vision and Mech-DLK SDK. Click here to view the details.