[發(fā)明專利]用于目標檢測的預訓練圖片合成方法和裝置

在審

• 申請?zhí)枺?/td>: 202110227613.9
• 申請日: 2021-03-01
• 公開（公告）號: CN112949631A
• 公開（公告）日: 2021-06-11
• 發(fā)明（設計）人: 張宇;李楊;張鴻
• 申請（專利權(quán)）人: 浙江大學
• 主分類號: G06K9/32
• 分類號: G06K9/32;G06K9/62
• 代理公司: 北京超凡宏宇專利代理事務所(特殊普通合伙) 11463
• 代理人: 梁韜

1.一種用于目標檢測的預訓練圖片合成方法，其特征在于，包括：

獲取帶目標位置標注的訓練圖片，在所述訓練圖片的目標位置生成至少一個候選框；

按照預設算法濾除所述目標位置上重疊的所述候選框，以及聚合相互接近的所述候選框；

根據(jù)預設單元尺寸對預設算法處理后的所述候選框進行尺寸規(guī)范化處理，獲得規(guī)范化候選框；

截取所述訓練圖片上所述規(guī)范化候選框的圖片塊，合成預訓練圖片。

2.根據(jù)權(quán)利要求1所述的用于目標檢測的預訓練圖片合成方法，其特征在于，所述按照預設算法濾除所述目標位置上重疊的所述候選框，以及聚合相互接近的所述候選框包括：

計算所述候選框與目標候選框重疊面積的比例值，濾除所述比例值大于預設比例值的所述候選框；

以剩余的所述候選框的中心點作為聚類點，采用層次聚類算法聚合相互接近的多個所述候選框。

3.根據(jù)權(quán)利要求2所述的用于目標檢測的預訓練圖片合成方法，其特征在于，所述比例值的計算算式包括：

式中，A()表示求面積，T_bbox表示目標候選框，R_bbox表示與所述目標候選框重疊的候選框，IoF為重疊的比例值。

4.根據(jù)權(quán)利要求1所述的用于目標檢測的預訓練圖片合成方法，其特征在于，所述根據(jù)預設單元尺寸對預設算法處理后的所述候選框進行尺寸規(guī)范化處理，獲得規(guī)范化候選框包括：

確定所述候選框的尺寸小于所述預設單元尺寸時，擴大所述候選框至所述預設單元尺寸；

確定所述候選框的尺寸大于所述預設單元尺寸時，擴大所述候選框至所述預設單元尺寸的整數(shù)倍。

5.根據(jù)權(quán)利要求4所述的用于目標檢測的預訓練圖片合成方法，其特征在于，所述規(guī)范化候選框的算式包括：

式中，h₂和w₂為規(guī)范化候選框的長度和寬度，h₁和w₁為規(guī)范化前候選框的長度和寬度，[]為取整運算，G為預設單元尺寸對應的像素值，H和W為所述訓練圖片的長度和寬度。

6.根據(jù)權(quán)利要求1所述的用于目標檢測的預訓練圖片合成方法，其特征在于，還包括：

確定所述預訓練圖片的尺寸大于預設尺寸時，根據(jù)所述預訓練圖片的最大邊與所述預設尺寸計算出縮放因子；

根據(jù)所述縮放因子對所述預訓練圖片進行縮小處理。

7.根據(jù)權(quán)利要求1所述的用于目標檢測的預訓練圖片合成方法，其特征在于，還包括：

對所述預訓練圖片進行歸一化處理后輸入至預設模型，對所述預設模型進行預訓練。

8.一種用于目標檢測的預訓練圖片合成裝置，其特征在于，包括：

候選框生成模塊，用于獲取帶目標位置標注的訓練圖片，在所述訓練圖片的目標位置生成至少一個候選框；

候選框濾除模塊，用于按照預設算法濾除所述目標位置上重疊的所述候選框，以及聚合相互接近的所述候選框；

候選框規(guī)范模塊，用于根據(jù)預設單元尺寸對預設算法處理后的所述候選框進行尺寸規(guī)范化處理，獲得規(guī)范化候選框；

用于目標檢測的預訓練圖片合成模塊，用于截取所述訓練圖片上所述規(guī)范化候選框的圖片塊，合成預訓練圖片。

9.一種計算機設備，其特征在于，包括存儲器以及處理器，所述存儲器存儲有計算機程序，所述計算機程序在所述處理器上運行時執(zhí)行根據(jù)權(quán)利要求1至7中任一項所述的用于目標檢測的預訓練圖片合成方法。

10.一種可讀存儲介質(zhì)，其特征在于，其存儲有計算機程序，所述計算機程序在處理器上運行時執(zhí)行權(quán)利要求1至7中任一項所述的用于目標檢測的預訓練圖片合成方法。