[发明专利]基于语义控制的生成数据集的方法、生成图像的系统

权利要求书

1.一种基于语义控制的房屋布局生成数据集的构建方法,其特征在于，所述方法包括：

根据预定的数据筛选规则从房屋布局图的原始数据集中筛选出合格数据和部分合格数据，删除不合格数据；

对所述合格数据进行半自动标注的图像分割与目标检测，获得房屋布局信息；

对所述部分合格数据进行手动标注的图像分割与目标检测，获得房屋布局信息；

所述房屋布局信息包括房间类型、房间坐标；

根据所述房屋布局信息，计算出房屋类型的数量、房间方位、房间相邻关系；

根据所述房屋类型的数量、房间方位、房间相邻关系，依据文字描述模板生成文字描述；

对所述文字描述再次过滤、规范化数据，生成最终目标检测数据集。

2.根据权利要求1所述的基于语义控制的房屋布局生成数据集的构建方法，其特征在于，所述房间类型包括：客厅LivingRoom、卧室Bedroom、阳台Balcony、厨房Kitchen、卫生间BathRoom和书房Library。

3.根据权利要求1所述的基于语义控制的房屋布局生成数据集的构建方法，其特征在于，所述数据筛选规则为：

房屋布局图中的房间有清晰的边界；

房屋布局图至少包含卧室Bedroom、厨房Kitchen、卫生间BathRoom房间类型；

房屋布局图是一个整体，且包含的房间是相互衔接的。

4.根据权利要求1所述的基于语义控制的房屋布局生成数据集的构建方法，其特征在于，依据文字描述模板生成文字描述的方法包括：

将每种房间类型的数量、房间方位、房间相邻关系，根据预先设定的文字描述模板，自动生成一段对应的文字描述，存放在excel表格中；

所述文字描述模板为每一行代表一个房屋，每一行的数据包含房屋编号、LivingRoom的数量、Bedroom的数量、Balcony的数量、Kitchen的数量、BathRoom的数量、Library的数量、房间方位、房间相邻关系。

5.根据权利要求1所述的基于语义控制的房屋布局生成数据集的构建方法，其特征在于，根据房屋布局信息，计算出房屋类型的数量、房间方位、房间相邻关系的方法包括：

根据房间类型，统计出每个房屋布局图中包含的每种房间类型的数量；

根据所述房间坐标点，判断该房间是矩形或多边形标注，然后计算该房间的中心点坐标，根据中心点坐标，判断该房间所在象限，从而判断所在房间方位；

对房屋布局图中的所有房间两两组合，并计算两个房间之间的曼哈顿距离，若曼哈顿距离小于两个房间外边界框的长度、宽度和的一半加上，则判定这两个房间是相邻关系。

6.根据权利要求1所述的基于语义控制的房屋布局生成数据集的构建方法，其特征在于，对所述合格数据进行半自动标注的图像分割与目标检测，获得房屋布局信息的方法包括：

使用目标检测模型对所述房屋布局图进行图像分割和目标检测，获取房屋布局图像的目标边界框列表及其对应的置信度得分列表；

对所述目标边界框列表及其对应的置信度得分列表进行非极大值抑制处理，获取每个所述房屋布局图中房间类型以及房间坐标点；

对房间类型列表中同类别的多个房间使用阿拉伯数字序号重命名。

7.根据权利要求6所述的基于语义控制的房屋布局生成数据集的构建方法，其特征在于：所述目标检测模型为MaskRCNN模型。

8.根据权利要求6所述的基于语义控制的房屋布局生成数据集的构建方法，其特征在于：所述目标检测模型为YOLACT模型。