[發明專利]一種防靜電鏈連接識別方法、裝置及可讀存儲介質

在審

• 申請號: 202210178514.0
• 申請日: 2022-02-24
• 公開（公告）號: CN114550061A
• 公開（公告）日: 2022-05-27
• 發明（設計）人: 許博;劉榮帥
• 申請（專利權）人: 蘇州浪潮智能科技有限公司
• 主分類號: G06V20/40
• 分類號: G06V20/40;G06V10/26;G06V10/25;G06V10/20;G06V10/764;G06K9/62;G06N3/04;G06N3/08
• 代理公司: 濟南舜源專利事務所有限公司 37205
• 代理人: 趙陽

1.一種防靜電鏈連接識別方法，其特征在于，包括如下步驟：

S1：接入視頻流，從視頻流中提取出檢測區域的圖片；

S2：將圖片輸入訓練后的圖目標檢測和分割模型，生成圖片的分割結果、目標框、目標框對應的類別和相應的得分；

S3：根據目標框獲取油罐車的位置和輪廓，以判斷圖片中是否包括有油罐車，若是，則轉到下一步，否則轉到步驟S1；

S4：根據分割結果、目標框和目標框對應的類別在圖片中確定分割后的油罐車和靜電釋放樁；

S5：根據圖片中分割后的油罐車的尺寸和靜電釋放樁，確定靜電釋放樁的位置和防靜電鏈的夾持位置，以在圖片中畫出目標提取區域；

S6：判斷目標提取區域中是否有其他遮擋目標；若是，轉到步驟S1；若否，轉到下一步；

S7：對目標提取區域中的圖片信息進行去噪處理；

S8：在目標提取區域中提取出防靜電鏈的邊緣信息；

S9：根據提取的邊緣信息判斷目標提取區域中是否有防靜電鏈；若有，則轉到步驟S11；若無，則轉到步驟S10；

S10：將連接計時清零，發出未連接防靜電鏈提示，并返回到步驟S1；

S11：查看連接計時是否開始，若未開始，則開始計時，并轉到下一步；若已開始，則直接轉到下一步；

S12：判斷是否到達規定的連接計時時間；若否，則轉到步驟S1；若是，則轉到下一步；

S13：發出防靜電鏈連接時間到達提示。

2.根據權利要求1所述的防靜電鏈連接識別方法，其特征在于，所述步驟S2包括：

S21：進行圖片預處理，取出圖片中的噪點，將圖片大小縮放到特定尺寸；

S22：將處理后的圖片輸入ResNet50網絡進行目標分類；

S23：將處理后的圖片FPN網絡，提取出特征圖；

S24：將提取的特征圖輸入RPN網絡得到圖片的目標推薦區域；

S25：將得到的特征圖和目標推薦區域同時輸入到ROI Align網絡，得到所需大小的特征圖；

S26：所需大小的特征圖經過分類層得到目標檢測區域和目標分割區域；

S27：目標分割區域經過兩次全連接層得到分割結果；

S28：目標檢測區域經過全連接層，分別得到目標框、分類和得分情況。

3.根據權利要求2所述的防靜電鏈連接識別方法，其特征在于，所述步驟S4具體為：

根據檢測目標區域和分割目標區域找到油罐車和靜電釋放樁。

4.根據權利要求3所述的防靜電鏈連接識別方法，其特征在于，所述步驟S5包括：

S51：根據圖片中分割后的油罐車的尺寸，識別出油罐車的類型；

S52：根據油罐車的類型識別出防靜電鏈的夾持位置；

S53：對靜電釋放樁位置進行標記；

S54：根據防靜電鏈的夾持位置和靜電釋放樁位置，在圖片中畫出目標提取區域。

5.根據權利要求1所述的防靜電鏈連接識別方法，其特征在于，所述圖目標檢測和分割模型的訓練過程，包括如下步驟：

S201：設置訓練的數據量，并提取預設訓練圖片作為訓練數據；

S202：進行圖片預處理，取出圖片中的噪點，將圖片大小縮放到特定尺寸；

S203：采用Mosaic算法和MixUp算法進行圖片的數據增強；

S204：將處理后的圖片輸入ResNet50網絡進行目標分類；

S205：將處理后的圖片FPN網絡，提取出特征圖；

S206：將提取的特征圖輸入RPN網絡得到圖片的目標推薦區域；

S207：將得到的特征圖和目標推薦區域同時輸入到ROI Align網絡，得到所需大小的特征圖；

S208：所需大小的特征圖經過分類層得到目標檢測區域和目標分割區域；

S209：目標分割區域經過兩次全連接層得到分割結果；

S210：目標檢測區域經過全連接層，分別得到目標框、分類和得分情況。