[发明专利]一种基于图像识别提取产品形状特征并计数的方法

说明书

本发明提供一种基于图像识别提取产品形状特征并计数的方法，包括以下步骤：S1：获取目标图片；S2：读取目标图片并将目标图片转化为灰度图；S3：对灰度图进行二值化处理，得到灰度图对应的二值化图像；S4：检测二值化图像轮廓，提取二值化图像轮廓的水平矩形坐标，通过坐标绘制矩形；S5：在绘制出的每个矩形左上角写入计数编号，根据编号最大值得到图形内产品个数；通过本发明提出的基于图像识别提取产品形状特征并计数的方法，每次容纳产品数量只受限于平铺场地，不需要提前维护产品形状参数，产品数量较大时，不需要多次将产品放入测量器具中，提高产品计数效率；不需要根据形状规则查询对应表得到产品数，节约时间。

技术领域

本发明涉及软件应用开发领域，尤其涉及一种基于图像识别提取产品形状特征并计数的方法。

背景技术

产品进行包装分拣时，工序有计数的要求。对于单体重量在10克以上的磁芯，通过称重反算个数是比较准确的计数方法，但是对于个重低于0.5克的细小磁芯类产品，用这种方式来计算误差较大。现有技术中，通过现场将产品紧密排列至量具内，通过查询排版数与数量对照表得到磁芯类产品的个数。

现有技术中对于克重较小的产品的计数方式存在如下弊端：

1、需要将产品排列紧密至量具内，根据形状规则查询对应表得到产品数，耗时耗力；

2、当产品数量较多时，量具大小有限，这个过程需要多次操作，效率低下。

发明内容

本发明的目的是为了解决现有技术中存在的缺点，而提出的一种基于图像识别提取产品形状特征并计数的方法。

为实现上述目的，本发明采用了如下技术方案：一种基于图像识别提取产品形状特征并计数的方法，包括以下步骤：

S1：获取目标图片；

工程师通过在线拍照或选择已有图片的方式获取目标图片；

所述目标图片中包含待计数产品且待计数产品平铺不重叠。

S2：读取目标图片并将目标图片转化为灰度图；

通过cv2.imread()函数读取目标图片的像素矩阵信息来读取目标图片；

所述像素矩阵信息包括每个像素的R、G、B值；

判断目标图片是否为灰度图；

如果目标图片的每一个像素所对应的R、G、B的值相等，则所述目标图片为是灰度图，进入步骤S3；

如果目标图片的每一个像素所对应的R、G、B的值不是完全相等，则所述目标图片不是灰度图，调用cv2.cvtColor()函数将目标图片转换成灰度图。

S3：对灰度图进行二值化处理，得到灰度图对应的二值化图像；

所述二值化图像是指只有黑色和白色两种颜色的图像，每个像素点可以用0或255表示，0表示黑色，255表示白色；

通过调用cv2.adaptiveThreshold()函数将灰度图转化为像素仅有0和255两个值的图像，将目标区域和背景区域彻底分割，得到灰度图对应的二值化图像；

所述目标区域是指图像中待计数产品所占区域，背景区域是指图像中除去待计数产品所占区域的其他区域；

每一个待计数产品对应一个目标区域。

S4：检测二值化图像轮廓，提取二值化图像轮廓的水平矩形坐标，通过坐标绘制矩形；

包括如下子步骤：

S41：检测二值化图像轮廓，找出图像中的轮廓值；

通过调用cv2.findContours()函数获取二值化图像中每个目标区域的轮廓及轮廓值，所述轮廓是一系列点的集合，轮廓值是指各个点的坐标值；

S42：根据图像的轮廓值，获取轮廓的垂直边界最小矩形；