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

在審

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

本發明提供一種用于目標檢測的預訓練圖片合成方法和裝置，該用于目標檢測的預訓練圖片合成方法包括：獲取帶目標位置標注的訓練圖片，在所述訓練圖片的目標位置生成至少一個候選框；按照預設算法濾除所述目標位置上重疊的所述候選框，以及聚合相互接近的所述候選框；根據預設單元尺寸對預設算法處理后的所述候選框進行尺寸規范化處理，獲得規范化候選框；截取所述訓練圖片上所述規范化候選框的圖片塊，合成預訓練圖片。本發明的用于目標檢測的預訓練圖片合成方法，通過利用帶目標位置標注的訓練圖片選取正樣本的候選框，進行候選框的濾除、聚合以及規范化后，可以合成規范的預訓練圖片，從而提高預訓練模型的精確度以及預訓練模型的預訓練效率。

技術領域

本發明涉及圖像識別技術領域，具體而言，涉及一種用于目標檢測的預訓練圖片合成方法、裝置、計算機設備和可讀存儲介質。

背景技術

現有的圖像識別模型在建立時會通過提前的預訓練過程來提高模型建立的效率與穩定性。但是現有的預訓練方法需要前置的圖像分類過程，該圖像分類過程比較費時，導致預訓練過程效率較低，從而導致整體的圖像識別模型在建立時效率較低。

發明內容

鑒于上述問題，本發明提供了一種用于目標檢測的預訓練圖片合成方法、裝置、計算機設備和可讀存儲介質，以提高預訓練模型的精確度以及預訓練模型的預訓練效率。

為了實現上述目的，本發明采用如下的技術方案：

一種用于目標檢測的預訓練圖片合成方法，包括：

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

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

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

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

優選地，所述的用于目標檢測的預訓練圖片合成方法中，所述按照預設算法濾除所述目標位置上重疊的所述候選框，以及聚合相互接近的所述候選框包括：

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

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

優選地，所述的用于目標檢測的預訓練圖片合成方法中，所述比例值的計算算式包括：

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

優選地，所述的用于目標檢測的預訓練圖片合成方法中，所述根據預設單元尺寸對預設算法處理后的所述候選框進行尺寸規范化處理，獲得規范化候選框包括：

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

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

優選地，所述的用于目標檢測的預訓練圖片合成方法中，所述規范化候選框的算式包括：

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

優選地，所述的用于目標檢測的預訓練圖片合成方法中，還包括：