[發明專利]一種基于圖像識別提取產品形狀特征并計數的方法

在審

• 申請號: 202211524389.0
• 申請日: 2022-12-01
• 公開（公告）號: CN115546462A
• 公開（公告）日: 2022-12-30
• 發明（設計）人: 楊松貴;黎冠軍
• 申請（專利權）人: 南京維拓科技股份有限公司
• 主分類號: G06V10/20
• 分類號: G06V10/20;G06V10/26;G06V10/28;G06V10/44;G06T7/00
• 代理公司: 南京科知維創知識產權代理有限責任公司 32270
• 代理人: 杜依民

本發明提供一種基于圖像識別提取產品形狀特征并計數的方法，包括以下步驟：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：根據圖像的輪廓值，獲取輪廓的垂直邊界最小矩形；