[發明專利]管狀物計數、模型的訓練方法、設備及存儲介質

在審

• 申請號: 202110410163.7
• 申請日: 2021-04-15
• 公開（公告）號: CN113066078A
• 公開（公告）日: 2021-07-02
• 發明（設計）人: 胡顯赫;張建鋒;王嘉
• 申請（專利權）人: 上海找鋼網信息科技股份有限公司
• 主分類號: G06T7/00
• 分類號: G06T7/00;G06T3/40;G06K9/42;G06K9/46;G06K9/62;G06N3/04;G06N3/08
• 代理公司: 北京超凡宏宇專利代理事務所(特殊普通合伙) 11463
• 代理人: 曹瑞敏

本申請提供一種管狀物計數、模型的訓練方法、設備及存儲介質，涉及圖像處理領域。該方法包括：采用所述圖像特征提取模塊，根據預設滑動窗口對應的縮放比例，對待檢測圖片進行縮放，得到所述待檢測圖片的特征圖；采用所述管狀物檢測模塊，對所述特征圖進行檢測，得到管狀物檢測信息；所述管狀物檢測信息包括：管狀物的數量；采用所述輸出模塊，輸出包括所述管狀物檢測信息的檢測結果。相對于現有技術，避免了管狀物計數統計人工成本和時間成本高，并且無法保證統計的準確率的問題。

技術領域

本申請涉及圖像處理技術領域，具體而言，涉及一種管狀物計數、模型的訓練方法、設備及存儲介質。

背景技術

鋼管、棒材等管狀物作為基礎建材的重要原材料之一，在工業界有著不可或缺的重要作用。管狀物按形狀可分為圓形、方形、矩形和異形管狀物。

由于鋼管、棒材等管狀物的種類繁多，用途廣泛，但卻也給生產環境帶來了困難；工地現場等生產環境中，管狀物堆積的數量存在多、雜、亂等缺點，人眼難以區分和計數，給生產人員、一線工人帶來極大困擾。一直以來，行業內默認是工人打計數的方法，以此來統計鋼管數量。

但是這樣的統計方法不但人工成本和時間成本高，并且無法保證統計的準確率。

發明內容

本申請的目的在于，針對上述現有技術中的不足，提供一種管狀物計數、模型的訓練方法、設備及存儲介質，以解決現有技術中管狀物計數統計人工成本和時間成本高，并且無法保證統計的準確率的問題。

為實現上述目的，本申請實施例采用的技術方案如下：

第一方面，本申請一實施例提供了一種管狀物計數方法，應用于管狀物計數模型，所述管狀物計數模型包括：圖像特征提取模塊、管狀物檢測模塊和輸出模塊；所述方法包括：

采用所述圖像特征提取模塊，根據預設滑動窗口對應的縮放比例，對待檢測圖片進行縮放，得到所述待檢測圖片的特征圖；

采用所述管狀物檢測模塊，對所述特征圖進行檢測，得到管狀物檢測信息；所述管狀物檢測信息包括：管狀物的數量；

采用所述輸出模塊，輸出包括所述管狀物檢測信息的檢測結果。

可選地，所述管狀物檢測信息還包括：各管狀物在所述特征圖中的位置；所述方法還包括：

采用所述管狀物檢測模塊，根據所述各管狀物在所述特征圖中的位置，在所述特征圖中對所述各管狀物進行標注；

采用所述輸出模塊輸出標注后的特征圖。

可選地，所述管狀物計數模型還包括：上采樣模塊；所述采用所述輸出模塊輸出標注后的特征圖之前，所述方法還包括：

采用所述上采樣模塊對所述標注后的特征圖進行上采樣處理；

所述采用所述輸出模塊輸出標注后的特征圖，包括：

采用所述輸出模塊輸出上采樣處理后的特征圖。

可選地，所述采用所述管狀物檢測模塊，根據所述各管狀物在所述特征圖中的位置，在所述特征圖中對所述各管狀物進行標注，包括：

采用所述管狀物檢測模塊，根據所述各管狀物在所述特征圖中的位置，在所述特征圖中所述各管狀物的位置處標注檢測框。

可選地，所述采用所述管狀物檢測模塊，根據所述各管狀物在所述特征圖中的位置，在所述特征圖中所述各管狀物的位置處標注檢測框，包括：

若所述特征圖中至少兩個檢測框的重合面積大于預設閾值，由所述管狀物檢測模塊，采用非極大值抑制算法從重合的檢測框中確定目標檢測框，并刪除所述重合的檢測框中所述目標檢測框之外的檢測框。