Classification

You are currently viewing the documentation for the latest version (2.2.0). To access a different version, click the "Switch version" button located in the upper-right corner of the page.

■ If you are not sure which version of the product you are currently using, please feel free to contact Mech-Mind Technical Support.

Function

Use the Classification model package to run inference on the input image, determine the class it belongs to, and output the corresponding label. This model package is applicable to data scenarios that do not require defect localization and where all categories can be enumerated.

Applicable to industries such as steel and machinery for distinguishing the front and back sides, orientations, types, and other attributes of various components.

intro

Input and Output

After importing the model package in the Deep Learning Model Package Inference Step, the following input and output ports will be displayed.

Input

Input port Data type Description

Image

Image/Color

Image input to this port will be used for deep learning model package inference.

Output

Output port Data type Description

Visualization Output

Image/Color

Visualized results.

Classification Confidence

Number[]

Confidence of the classification result.

Classification Label

String[]

Labels for classification results.

Bounding Box

Shape2D/Rectangle[]

Min bounding box.

Parameter Description

The following parameters need to be adjusted when the classification model package is imported into this Step.

Model Package Settings

Parameter Description

Model Manager Tool

Parameter description: This parameter is used to open the deep learning model package management tool and import the deep learning model package. The model package file is a “.dlkpack” file exported by Mech-DLK.
Tuning instruction: Please refer to Deep Learning Model Package Management Tool for the usage.

Model Name

Parameter description: This parameter is used to select the model package that has been imported for this Step.
Tuning instruction: After importing a deep learning model package with the Deep Learning Model Package Management Tool, select the corresponding model package name from the drop-down list.

Release Original Model Package After Switching

Parameter description: This parameter determines whether to release the resources occupied by the original model package immediately when the model package is switched.
Default setting: Selected.
Instruction: Once this option is selected, when the Step switches to another model package, the system will immediately release the original model package resources, even if the model package is still in use by other Steps. When this option is not selected, the system will automatically release the original model package only when it is no longer used by any Steps.

Model Package Type

Parameter description: Once a Model Name is selected, the DI Algo Type Translated String will be filled automatically.

Input Batch Size

Parmeter description: The number of images processed during each inference.

GPU ID

Parameter description: This parameter is used to select the device ID of the GPU that will be used for the inference.
Tuning instruction: Once you have selected the model name, you can select the GPU ID in the drop-down list of this parameter.

Pre-Process

Parameter Description

ROI File

Parameter description: This parameter is used to set or modify the ROI of the input image.

Tuning instruction: Once the deep learning model is imported, a default ROI will be applied. If you need to edit the ROI, click Open the editor. Edit the ROI in the pop-up Set ROI window, and fill in the ROI name.

Before the inference, please check whether the ROI set here is consistent with the one set in Mech-DLK. If not, the recognition result may be affected.

During the inference, the ROI set during model training, i.e. the default ROI, is usually used. If the position of the object changes in the camera’s field of view, please adjust the ROI.

If you would like to use the default ROI again, please delete the ROI file name below the Open the editor button.

Post-Process

Parameter Description

Inference Configuration

Parameter description: This parameter is used to configure parameters related to classification model package inference. You can click Open the editor to open the inference configuration window.
Instruction: Please refer to Inference Configuration Tool for detailed parameter description.

Class Display Mode

Parameter description: Select to display classes in the output results by name or by index.

Visualization Settings

Parameter Description

Draw result on image

Parameter description: Once enabled, the detection result will be displayed on the image.
Default value: Disabled.
Tuning recommendation: Please set this parameter according to your actual needs.

Customize Font Size

Parameter description: This parameter determines whether to customize the font size in the visualized outputs. Once this option is selected, you should set the Font Size (0–10).
Default value: Disabled.
Tuning recommendation: Please set this parameter according to your actual needs.

Font Size (0–10)

Parameter description: This parameter is used to set the font size in the visualized outputs.
Default value: 1.5
Tuning recommendation: Please set this parameter according to your actual needs.

Is this page helpful?

You can give a feedback in any of the following ways:

We Value Your Privacy

We use cookies to provide you with the best possible experience on our website. By continuing to use the site, you acknowledge that you agree to the use of cookies. If you decline, a single cookie will be used to ensure you're not tracked or remembered when you visit this website.

  • Preference
  • I accept
  • I don't accept