对象提取方法及装置制造方法及图纸

本申请实施例提供一种对象提取方法及装置，该方法包括：将包括至少一个对象的第一图片输入至语义分割特征网络，以使得语义分割特征网络输出第一特征图。根据语义分割分类网络对第一特征图进行处理，得到语义分割结果图，语义分割结果图中包括对象掩码图的图层和背景掩码图的图层。根据兴趣点生成网络对第一特征图和对象掩码图的图层进行处理，得到至少一个兴趣点。根据兴趣点控制网络、实例选择网络对第一特征图和至少一个兴趣点进行处理，得到至少一个兴趣点各自对应的目标掩码图。根据至少一个兴趣点各自对应的目标掩码图，分别确定各个对象在第一图片中的位置，以提取至少一个对象。本实施例有效保证了提取的杆状物对象的效果和准确性。效果和准确性。效果和准确性。

【技术实现步骤摘要】
对象提取方法及装置


[0001]本申请实施例涉及计算机技术，尤其涉及一种对象提取方法及装置。

技术介绍

[0002]在地图建图的过程中，通常需要提取图像信息中的杆状物对象，例如路灯杆、电线杆等。
[0003]目前，相关技术在对图像信息中的杆状物对象进行提取时，可以首先检测出图像中的单个目标的包围框的边界，之后再采用分割方法完成对前景和背景的语义分割，最终结合单个目标的包围框的边界以及前景背景语义分割的结果，以实现提取出图像中的各个杆状物。
[0004]然而，由于杆状物对象的形状是细长的，其包围框长宽比例和大小不均衡，因此采用上述包围框检测的方式来进行对象的提取的准确性较差。

技术实现思路

[0005]本申请实施例提供一种对象提取方法及装置，以克服对象的提取的准确性较差的问题。
[0006]第一方面，本申请实施例提供一种对象提取方法，包括：
[0007]将包括至少一个对象的第一图片输入至语义分割特征网络，以使得所述语义分割特征网络输出第一特征图，其中，所述第一特征图用于指示所述第一图片的整体特征；<本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.一种对象提取方法，其特征在于，包括：将包括至少一个对象的第一图片输入至语义分割特征网络，以使得所述语义分割特征网络输出第一特征图，其中，所述第一特征图用于指示所述第一图片的整体特征；根据语义分割分类网络对所述第一特征图进行处理，得到语义分割结果图，所述语义分割结果图中包括对象掩码图的图层和背景掩码图的图层；根据兴趣点生成网络对所述第一特征图和所述对象掩码图的图层进行处理，得到至少一个兴趣点，其中，一个兴趣点对应所述第一图片中的一个对象；根据兴趣点控制网络、实例选择网络对所述第一特征图和所述至少一个兴趣点进行处理，得到所述至少一个兴趣点各自对应的目标掩码图，其中，所述目标掩码图用于指示所述兴趣点对应的对象在所述第一图片中的位置；根据所述至少一个兴趣点各自对应的目标掩码图，分别确定各个对象在所述第一图片中的位置，以提取所述至少一个对象。2.根据权利要求1所述的方法，其特征在于，所述根据兴趣点生成网络对所述第一特征图和所述对象掩码图的图层进行处理，得到至少一个兴趣点，包括：将所述第一特征图输入至兴趣点生成网络，以使得所述兴趣点生成网络输出第二特征图，其中，所述第二特征图用于指示所述第一图片中的各像素点作为兴趣点的概率；根据所述对象掩码图的图层和第一预设阈值，将所述第二特征图中非对象的像素点以及作为兴趣点的概率小于所述第一预设阈值的像素点的值设置为0，得到处理后的第二特征图；对所述处理后的第二特征图中的各像素点进行遍历，若当前遍历到的像素点的值大于其周围的M
×
N的范围内的其余像素点的值，则将当前遍历到的像素点确定为兴趣点，其中，所述M为大于等于1的整数，所述N为大于等于1的整数。3.根据权利要求2所述的方法，其特征在于，所述根据兴趣点控制网络、实例选择网络对所述第一特征图和所述至少一个兴趣点进行处理，得到所述至少一个兴趣点各自对应的目标掩码图，包括：根据所述兴趣点控制网络对所述第一特征图和所述至少一个兴趣点进行处理，得到各所述兴趣点分别对应的方差特征和均值特征；根据所述实例选择网络对所述第一特征图、所述方差特征、所述均值特征进行处理，得到所述至少一个兴趣点各自对应的目标掩码图。4.根据权利要求3所述的方法，其特征在于，所述根据所述兴趣点控制网络对所述第一特征图和所述至少一个兴趣点进行处理，得到各所述兴趣点分别对应的方差特征和均值特征，包括：将所述第一特征图输入至兴趣点控制网络，以使得所述兴趣点控制网络输出第三特征图，其中，所述第三特征图用于指示所述兴趣点的数据分布特征；针对所述N个兴趣点中的任一个第一兴趣点，根据第一算法对所述第三特征图和所述第一兴趣点进行处理，得到第四特征图，其中，所述第四特征图包括所述第一兴趣点的坐标在所述第三特征图中对应的特征；将所述第四特征图输入至第一全连接卷积层，得到所述第一兴趣点的第一控制值，以及，将所述第四特征图输入至第二全连接卷积层，得到所述第二兴趣点的第二控制值，其
中，所述第一控制值用于指示所述兴趣点的方差特征，所述第二控制值用于指示所述兴趣点的均值特征。5.根据权利要求4所述的方法，其特征在于，所述根据所述实例选择网络对所述第一特征图、所述方差特征、所述均值特征进行处理，得到所述至少一个兴趣点各自对应的目标掩码图，包括：将所述第一特征图输入至实例选择网络，以使得所述实例选择网络输出第五特征图，其中，所述第五特征图用于指示所述至少一个对象在所述第一图像中的语义信息特征；对所述第五特征图进行归一化操作，并针对所述N个兴趣点中的任一个第一兴趣点，将归一化操作后的第五特征图和所述第一兴趣点的第一控制值做乘法操作，以...

【专利技术属性】
技术研发人员：黄伟杰，田文善，康勇，
申请(专利权)人：北京四维图新科技股份有限公司，
类型：发明
国别省市：