[发明专利]一种基于图像生成技术的垃圾物品图像数据集构建方法

摘要

本发明涉及一种基于图像生成技术的垃圾物品图像数据集构建方法：搜集整理单目标垃圾图像数据集；利用U2‑Net模型得到单个垃圾物品的mask图像；对单个垃圾物品进行抠图，构成Crop数据集；从Crop数据集中随机抽取多张图像，依次随机粘贴在一张大的纯色背景图上；合成多张图像构成Syn数据集；将采集的少量多目标真实图视为目标域，Syn数据集视为源域，用CycleGAN模型进行风格转换；对风格转换结果进行优化处理，得到最终合成图。本发明的一种基于图像生成技术的垃圾物品图像数据集构建方法，将繁杂的数据采集和标注过程交给计算机自动完成，可显著提高数据集的构建效率，降低人工成本。