Tag Extraction Tool

Decompress the segmentation dataset, remove the defective images from the images folder of the segmentation dataset, and run the ./dataset_tools/tag_extract/tag_paste_cls_dataset.py script to perform workpiece alignment on the segmented images to obtain images for classification.

python3 ./dataset_tools/tag_extract/tag_paste_cls_dataset.py --input_dataset_path=/seg/dataset/path --output_dataset_path=/data/output/path
Table 1 Parameter description

Parameter

Type

Value Range

Mandatory/Optional

Description

--input_dataset_path

String

-

Mandatory

Path of the segmentation dataset.

--output_dataset_path

String

-

Optional. By default, output is created in the segmentation dataset directory.

Path of the output images.

The script creates an images folder in output_dataset_path and saves the generated image data to images.