公开(公告)号：CN117011639A

公开(公告)日：2023-11-07

申请号：CN202310786726.1

申请日：2023-06-29

Applicant: 福建南方路面机械股份有限公司 ,  华侨大学

Inventor： 杨建红 ,  张宝裕 ,  李腾 ,  黄文景 ,  房怀英 ,  王英俊 ,  曹礼禧 ,  杨坤

IPC: G06V10/774 ,  G06V10/764 ,  G06V10/44

Abstract: 本发明提供基于SAM分割模型的固废数据集生成方法及装置，方法包括：步骤S1、采集彩色的固废图像；步骤S2、通过SAM分割模型得到各固废图像中每个物体实例的轮廓信息；步骤S3、对各固废图像中的物体实例进行自动类别标注得到数据集标注文件；步骤S4、根据数据集标注文件中的物体轮廓信息，复制每个离散物体实例并构成固废池；步骤S5、复制每种堆叠组合所对应的物体实例，并随机堆叠位置和堆叠程度，将堆叠后的堆叠实例加入固废池内；步骤S6、选择粘贴的背景图像并生成分布点，随机在固废池中选择堆叠实例或者离散实例，粘贴到分布点的位置。本发明能够低成本、高效地生成堆叠工况下的带标注数据集。