[發明專利]電氣圖紙中帶有毛刺特征的表格信息提取方法及裝置

在審

• 申請號: 202211352828.4
• 申請日: 2022-11-01
• 公開（公告）號: CN115690823A
• 公開（公告）日: 2023-02-03
• 發明（設計）人: 陳中;譚林林;曹衛國;錢晶
• 申請（專利權）人: 南京云階電力科技有限公司
• 主分類號: G06V30/422
• 分類號: G06V30/422;G06V30/412;G06V10/28;G06V10/32;G06V10/34;G06V10/44;G06V10/48;G06V10/764;G06V10/82;G06N3/0464;G06N3/08;G06T7/73
• 代理公司: 北京眾允專利代理有限公司 11803
• 代理人: 宋曉韻

1.一種電氣圖紙中帶有毛刺特征的表格信息提取方法，其特征在于包括如下步驟：

S1：通過目標檢測算法，檢測表格所在區域；

S2：將識別結果保留坐標，對原圖進行坐標裁剪，提取表格所在區域；

S3：批量讀取圖片，將圖片灰度化處理后進行二值化，腐蝕和膨脹操作，利用霍夫線變換函數檢測橫豎線；

S4：對橫豎線組合的區域檢測物體的輪廓；

S5：根據輪廓坐標獲取最大表格，通過剪切把最外面的毛刺去掉，根據輪廓索引分別提取去毛刺后表格內的所有單元格。

2.如權利要求1所述的電氣圖紙中帶有毛刺特征的表格信息提取方法，其特征在于，所述步驟S1具體包括：

S101：通過AutoCAD根據實際電氣工程要求設計繪制生成PDF，再轉化為計算機可以處理的圖像格式；

S102：對圖像中的表格所在區域進行標注并訓練出模型權重，將此權重用于檢測電氣設計圖紙中的表格區域；

所述步驟S2具體包括：

S201：通過YOLOv5算法中設置save-txt參數，將檢測出的表格區域坐標保留在TXT文檔中；

S202：將YOLO格式TXT標注文件轉換為VOC格式XML標注文件；

S203：通過OpenCV讀取XML中的信息，在原圖中進行裁剪，提取并保存表格區域所在圖片。

3.如權利要求1所述的電氣圖紙中帶有毛刺特征的表格信息提取方法，其特征在于，所述步驟S3具體包括如下步驟:

S301：批量讀取圖片，使用cv2.cvtColor將圖片進行灰度化處理；

S302：將灰度化后的圖片使用cv2.threshold進行二值化；

S303：將二值化后的圖片使用cv2.erode進行腐蝕操作，iterations＝1腐蝕的次數為1；

S304：將腐蝕后的圖片使用cv2.dilate進行膨脹操作，iterations＝1膨脹的次數為1；

S305：使用霍夫變換cv2.HoughLinesP檢測橫豎線。

4.如權利要求1所述的電氣圖紙中帶有毛刺特征的表格信息提取方法，其特征在于，所述步驟S4具體包括如下步驟：

S401：使用cv2.findContours函數對橫豎線區域檢測物體的輪廓；

S402：使用cv2.boundingRect函數輸出檢測到輪廓的坐標；

S403：將檢測出的輪廓坐標保留在空列表，并使用sorted進行坐標從小到大排序，刪除最外框的無效坐標。

5.如權利要求1所述的電氣圖紙中帶有毛刺特征的表格信息提取方法，其特征在于，所述步驟S5具體包括如下步驟：

S501：根據剩余坐標中的最大坐標使用cv2.rectangle函數在邊緣畫出線的粗細為5的矩形；

S502：根據矩形坐標進行切割圖片，若表格為單行，則擴大切割范圍；

S503：根據輪廓索引分別提取去毛刺后表格內的所有單元格。

6.一種電氣圖紙中帶有毛刺特征的表格信息提取裝置，其特征在于包括：

表格區域識別模塊(1)：用于識別圖紙中的表格區域，并將提取出的表格所在區域另存為新的目標圖片；

圖像預處理模塊(2)：用于將目標圖片轉換為灰度圖，設定閾值對其進行二值化處理，之后通過腐蝕膨脹對其使用霍夫變換檢測橫豎線；

輪廓檢測模塊(3)：用于檢測圖片中物體的輪廓，并保留所有輪廓坐標；

坐標排序及刪除模塊(4)：用于對輪廓坐標進行從小到大排序并刪除最大的無效坐標；

表格去毛刺模塊(5)：用于根據剩余坐標中的最大坐標畫出虛擬矩形，裁剪掉虛擬矩形外的多余毛刺。