[发明专利]一种图像主体提取方法及装置

说明书

技术领域

本发明涉及图像处理技术领域，尤其涉及一种图像主体提取方法及装置。

背景技术

近年来网购平台取得了长足的发展，网购平台积累了大量的商品图像信息，如何更有效地实现对商品图像信息的组织、分析、检索和向消费者展示已经变得十分重要。商品图像的内容包括商品主体和背景，当用户上传一幅商品图像并期望搜索与该图相同或相似的商品时，用户更关注于商品本身，因此，提取商品图像中的商品主体成为非常重要的一项工作。目前，图像主体提取方法通常基于Graph cuts算法，Graph cuts算法的主要工作流程为：在图像四个角上设置背景种子点，执行Graph cuts算法得到主体mask，采用滑动窗口进行检测，获取图像主体的边距并得到主体外截矩形。但是，Graph cuts算法按像素分割，执行效率低，并且，固定设置背景种子点的做法使得主体占整个画幅较大且与背景种子点重叠时无法成功分割。

发明内容

有鉴于此，本发明要解决的一个技术问题是提供一种图像主体提取方法及装置。

根据本发明的一个方面，提供一种图像主体提取方法，包括：基于边缘检测算子对待处理图像进行边缘检测处理，得到边缘检测后的图像；生成检测滑窗，采用所述检测滑窗在所述边缘检测后的图像中进行滑窗检测；根据所述检测滑窗内的像素点数量以及图像主体识别规则在所述边缘检测后的图像中进行图像主体区域提取处理，将提取出的图像作为所述待处理图像的主体部分。

可选地，获取所述待处理图像中包含的像素点的颜色值，根据所述像素点的颜色值以及图像纯色判断规则确定所述待处理图像是否为纯色图像；如果是，则基于所述边缘检测算子对所述待处理图像进行边缘检测处理；如果否，则结束对所述待处理图像的处理。

可选地，所述获取所述待处理图像中包含的像素点的颜色值、根据所述像素点的颜色值以及图像纯色判断规则确定所述待处理图像是否为纯色图像包括：在所述待处理图像的外围设置检测界限，确定所述检测界限内的图像检测区域；获取所述图像检测区域中包含的像素点的颜色值，统计具有相同颜色值的像素点在所述图像检测区域中包含的全部像素点中的占比；获取所述占比最高的颜色值以及占比值，判断所述占比值是否高于占比阈值；如果是，则确定所述待处理图像为纯色图像。

可选地，所述生成检测滑窗、采用所述检测滑窗在所述边缘检测后的图像中进行滑窗检测包括：在所述边缘检测后的图像的边沿处分别设置水平滑窗和垂直滑窗；控制所述水平滑窗、所述垂直滑窗分别向所述边缘检测后的图像的中心滑动，用以确定所述图像主体区域的边界。

可选地，所述水平滑窗的高度像素数为所述边缘检测后的图像的宽度像素数，所述水平滑窗的宽度像素数为t1；所述垂直滑窗的宽度像素数为所述边缘检测后的图像的长度像素数，所述垂直滑窗的高度像素数为t2；其中，控制所述水平滑窗、所述垂直滑窗分别以步长t1和t1向所述边缘检测后的图像的中心滑动。

可选地，在所述边缘检测后的图像的边沿处设置所述水平滑窗、所述垂直滑窗时，在所述水平滑窗、所述垂直滑窗中去除干扰区域，其中，所述干扰区域包括：商标区域。

可选地，所述根据所述检测滑窗内的像素点数量以及图像主体识别规则在所述边缘检测后的图像中进行图像主体区域提取处理包括：在所述水平滑窗、所述垂直滑窗向所述边缘检测后的图像的中心滑动的过程中，分别获取所述水平滑窗、所述垂直滑窗中所包含的像素数与在上一个滑动位置的所述水平滑窗、所述垂直滑窗中所包含的像素数的差值d1和d2；计算连续获取的两个d1的差值dk1，当确定dk1大于第一像素阈值时，则停止所述水平滑窗滑动并确定所述图像主体区域的垂直边界；计算连续获取的两个d2的差值dk2，当确定dk2大于第二像素阈值时，则停止所述垂直滑窗滑动并确定所述图像主体区域的水平边界。

可选地，在基于所述边缘检测算子对所述待处理图像进行边缘检测处理之前，采用平滑滤波器对对所述待处理图像进行滤波处理；其中，所述边缘检测算子包括：Canny算子。

根据本发明的又一方面，提供一种图像主体提取装置，包括：边缘检测模块，用于基于边缘检测算子对待处理图像进行边缘检测处理，得到边缘检测后的图像；滑窗检测模块，用于生成检测滑窗，采用所述检测滑窗在所述边缘检测后的图像中进行滑窗检测；主体提取模块，用于根据所述检测滑窗内的像素点数量以及图像主体识别规则在所述边缘检测后的图像中进行图像主体区域提取处理，将提取出的图像作为所述待处理图像的主体部分。