昇腾社区首页
中文
注册

标签提取工具

解压分割数据集,从分割的数据集“images”文件夹中,去除有缺陷的图片,执行脚本“./dataset_tools/tag_extract/tag_paste_cls_dataset.py”对分割图片进行工件对齐操作,拿到分类图片。

python3 ./dataset_tools/tag_extract/tag_paste_cls_dataset.py --input_dataset_path=/seg/dataset/path --output_dataset_path=/data/output/path
表1 参数说明

参数名

类型

取值范围

是否必选

说明

--input_dataset_path

str

-

必选。

分割数据集文件夹路径。

--output_dataset_path

str

-

可选,默认在分割数据集目录下新建output。

输出图片路径。

脚本会在“output_dataset_path”目录下新建“images”文件夹,并将生成的图片数据放在“images”中。