COCO类型目标检测数据集离线数据增强

增强工具路径“./dataset_tools/coco_data_aug/coco_data_aug.py”，目标检测数据增强内容支持离线数据增强，其中有移动性和遮挡两类增强方式。

coco_data_aug的参数名、类型、取值范围、默认值及说明如表1所示。

表1 coco_data_aug参数说明
参数名	类型	取值范围	默认值	说明
--input_dataset_path	str	-	“”	coco数据集路径。
--output_dataset_path	str	-	./output_dataset_path	增强后数据保存路径。
--x_offset	bool	True 或 False	False	X轴移动增强开关。
--y_offset	bool	True 或 False	False	Y轴移动增强开关。
--offset_target_times	int	[1, 10]	1	移动增强数据增加的倍数。
--obj_block	bool	True 或 False	False	遮挡增强开关。
--block_labels	str	-	label0,label1	具体遮挡目标标签列表，多个标签之间以英文逗号分隔。
--obj_erase	bool	True 或 False	False	遮挡之擦除增强开关。

默认参数示例如下：

python3 ./dataset_tools/coco_data_aug/coco_data_aug.py --input_dataset_path=./train --output_dataset_path=./output_dataset_path --x_offset=True --y_offset=True --obj_block=True --block_labels='label0,label1' --obj_erase=True --offset_target_times=2

各类增强模式可以选择组合或者单独开启，如果开启遮挡增强开关，则需要在“--block_labels”参数中指定具体的被遮挡标签。

执行命令参考log信息如下：

图1 命令执行后log信息

脚本执行完毕后会在“--output_dataset_path”参数指定的输出目录中生成增强后的数据。

图2 “--output_dataset_path”参数指定目录

父主题： 数据集处理工具包