[發明專利]一種輸配電設備圖像樣本庫的構建方法及裝置

在審

• 申請號: 202011364973.5
• 申請日: 2020-11-27
• 公開（公告）號: CN112528055A
• 公開（公告）日: 2021-03-19
• 發明（設計）人: 王繼業;高靈超;李浩松;王路濤;陳振宇;郝赫;郭敬林
• 申請（專利權）人: 國家電網有限公司大數據中心;北京國電通網絡技術有限公司
• 主分類號: G06F16/51
• 分類號: G06F16/51;G06K9/62;G06T3/40;G06Q50/06;G06N3/04
• 代理公司: 北京風雅頌專利代理有限公司 11403
• 代理人: 李翔

1.一種輸配電設備圖像樣本庫的構建方法，其特征在于，包括：

建立信息記錄表；

獲取原始圖像；

通過哈希函數算法將所述原始圖像轉換為哈希值；

將所述原始圖像輸入到預訓練好的目標檢測模型中，獲得缺陷圖像以及所述缺陷圖像中的缺陷信息；

將所述哈希值及缺陷信息記錄在所述信息記錄表中；

通過圖像縮放算法，將所述原始圖像縮小為相應的縮略圖像；

建立樣本庫，所述樣本庫包括所述原始圖像、縮略圖像及信息記錄表。

2.根據權利要求1所述的輸配電設備圖像樣本庫的構建方法，其特征在于，所述哈希函數算法為均值哈希算法。

3.根據權利要求2所述的輸配電設備圖像樣本庫的構建方法，其特征在于，所述均值哈希算法包括：

將所述原始圖像縮放為8*8，共64個像素的網格圖；

將所述網格圖轉化為64級的灰度圖；

計算所述灰度圖中所有像素灰度值的平均值；

將所述灰度圖中64個像素的灰度值與平均值依次進行比較，灰度值大于平均值的像素，灰度值計為1，否則計為0；

將所述64個像素的灰度值依次組合為64位的整數，即為哈希值。

4.根據權利要求1所述的輸配電設備圖像樣本庫的構建方法，其特征在于，所述目標檢測模型為YOLOv3或者YOLOv4模型；

所述目標檢測模型的訓練方法為：

獲取目標檢測訓練集，所述目標檢測訓練集包括標記好缺陷區域坐標以及缺陷信息的訓練圖像；

將所述目標檢測訓練集內的訓練圖像輸入初始的YOLOv3或者YOLOv4模型進行迭代運算，直至達到預設的迭代次數；

獲得訓練好的YOLOv3或者YOLOv4模型。

5.根據權利要求1所述的輸配電設備圖像樣本庫的構建方法，其特征在于，所述圖像縮放算法為基于內容感知的圖像縮放算法。

6.根據權利要求5所述的輸配電設備圖像樣本庫的構建方法，其特征在于，所述基于內容感知的圖像縮放算法包括：

提取所述灰度圖的融合直線信息的邊緣圖；

提取所述灰度圖的梯度圖；

利用HFPM算法處理所述缺陷圖像、邊緣圖和梯度圖，融合得到能量函數；

依據所述能量函數，使用線裁剪算法對所述原始圖像進行裁剪；

得到處理后的縮略圖像。

7.一種輸配電設備圖像樣本庫的構建裝置，其特征在于，包括：

第一建立模塊，用于建立信息記錄表；

第一獲取模塊，用于獲取原始圖像；

第一計算模塊，用于通過哈希函數算法將所述原始圖像轉換成哈希值；

第二計算模塊，用于將所述原始圖像輸入到預訓練好的目標檢測模型中，獲得缺陷圖像以及所述缺陷圖像中的缺陷信息；

第一記錄模塊，用于將所述哈希值缺陷信息記錄在信息記錄表中；

第三計算模塊，用于通過圖像縮放算法，將所述原始圖像縮小為相應的縮略圖像；

第二建立模塊，用于建立樣本庫，所述樣本庫包括所述原始圖像、縮略圖像及信息記錄表。

8.根據權利要求5所述的輸配電設備圖像樣本庫的構建裝置，其特征在于，所述哈希函數算法為均值哈希算法，所述第一計算模塊包括：

網格圖轉化單元，用于將所述原始圖像縮放為8*8，共64個像素的網格圖；

灰度圖轉化單元，用于將所述網格圖轉化為64級的灰度圖；

灰度值計算單元，用于計算所述灰度圖中所有像素灰度值的平均值；

灰度值比較單元，用于將所述灰度圖中64個像素的灰度值與平均值依次進行比較，灰度值大于平均值的像素，灰度值計為1，否則計為0；

哈希值獲得單元，用于將所述64個像素的灰度值依次組合為64位的整數，即為哈希值。