[發明專利]一種錨框參數的確定方法、裝置、設備及可讀存儲介質

有效

• 申請號: 202110933981.5
• 申請日: 2021-08-16
• 公開（公告）號: CN113378864B
• 公開（公告）日: 2021-11-12
• 發明（設計）人: 李一清;何林陽;王威;宋志龍;周凱;程愉
• 申請（專利權）人: 浙江啄云智能科技有限公司
• 主分類號: G06K9/46
• 分類號: G06K9/46;G06K9/62;G06N3/04;G06N3/08
• 代理公司: 北京樂知新創知識產權代理事務所(普通合伙) 11734
• 代理人: 江宇

1.一種錨框參數的確定方法，其特征在于，包括：

獲取圖像，對圖像進行預處理，所述圖像包括多個目標標注框；

根據輸出特征圖的跨度、預處理后圖像的第一尺寸和預設的錨框參數生成各特征圖的錨框；所述根據輸出特征圖的跨度、各所述特征圖的第二尺寸及預設的錨框參數生成各所述特征圖的錨框的步驟包括：根據輸出特征圖的跨度及預設的錨框參數生成各所述特征圖對應的基礎錨框；根據各所述特征圖的第二尺寸及所述基礎錨框生成各所述特征圖的錨框；錨框參數為錨框尺度和錨框高寬比；輸出特征圖的跨度為輸入圖像與特征圖的尺寸之間的比值；

根據多個所述目標標注框從多個所述錨框中選擇出作為正樣本的多個目標錨框，得到所述目標錨框與所述目標標注框的多個匹配對；

根據所述目標錨框與所述目標標注框的多個匹配對計算損失；

根據預設規則對所述錨框參數進行更新，并返回獲取圖像，對圖像進行預處理，所述圖像包括多個目標標注框的步驟，直至所述損失收斂，得到目標錨框參數。

2.根據權利要求1所述的錨框參數的確定方法，其特征在于，所述根據輸出特征圖的跨度、預處理后圖像的第一尺寸和預設的錨框參數生成各特征圖的錨框的步驟包括：

根據輸出特征圖的跨度和預處理后圖像的第一尺寸計算各特征圖的第二尺寸；

根據輸出特征圖的跨度、各所述特征圖的第二尺寸和預設的錨框參數生成各所述特征圖的錨框。

3.根據權利要求1所述的錨框參數的確定方法，其特征在于，所述根據輸出特征圖的跨度及預設的錨框參數生成各所述特征圖對應的基礎錨框的步驟包括：

根據輸出特征圖的跨度及預設的錨框參數計算各所述特征圖對應的基礎錨框參數；

根據各所述基礎錨框參數生成各所述特征圖的基礎錨框。

4.根據權利要求1所述的錨框參數的確定方法，其特征在于，所述根據所述目標錨框與所述目標標注框的多個匹配對計算損失的步驟包括：

計算所述目標錨框與所述目標標注框的每個匹配對的GIoU損失；

根據每個匹配對的GIoU損失計算GIoU損失均值及GIoU損失標準差；

將所述GIoU損失均值與所述GIoU損失均值的第一權重的第一乘積、所述GIoU損失標準差與所述GIoU損失標準差的第二權重的第二乘積進行累加，得到所述損失。

5.一種錨框參數的確定裝置，其特征在于，包括：

獲取單元，用于獲取圖像，對圖像進行預處理，所述圖像包括多個目標標注框；

生成單元，用于根據輸出特征圖的跨度、預處理后圖像的第一尺寸和預設的錨框參數生成各特征圖的錨框；所述生成子單元，包括：第一生成子單元，用于根據輸出特征圖的跨度及預設的錨框參數生成各所述特征圖對應的基礎錨框；第二生成子單元，用于根據各所述特征圖的第二尺寸及所述基礎錨框生成各所述特征圖的錨框；錨框參數為錨框尺度和錨框高寬比；輸出特征圖的跨度為輸入圖像與特征圖的尺寸之間的比值；

選擇單元，用于根據多個所述目標標注框從多個所述錨框中選擇出作為正樣本的多個目標錨框，得到所述目標錨框與所述目標標注框的多個匹配對；

計算單元，用于根據所述目標錨框與所述目標標注框的多個匹配對計算損失；

更新單元，用于根據預設規則對錨框參數進行更新，并返回獲取圖像，對圖像進行預處理，所述圖像包括多個目標標注框的步驟，直至所述損失收斂，得到目標錨框參數。

6.根據權利要求5所述的錨框參數的確定裝置，其特征在于，所述生成單元包括：

計算子單元，用于根據輸出特征圖的跨度和預處理后圖像的第一尺寸計算各特征圖的第二尺寸；

生成子單元，用于根據輸出特征圖的跨度、各所述特征圖的第二尺寸和預設的錨框參數生成各所述特征圖的錨框。

7.一種電子設備，其特征在于，包括：

至少一個處理器；以及與所述至少一個處理器通信連接的存儲器；其中，所述存儲器存儲有可被所述至少一個處理器執行的指令，所述指令被所述至少一個處理器執行，以使所述至少一個處理器執行如權利要求1-4任意一項所述的錨框參數的確定方法。

8.一種計算機可讀存儲介質，其特征在于，所述計算機可讀存儲介質存儲有計算機指令，所述計算機指令用于使計算機執行如權利要求1-4任意一項所述的錨框參數的確定方法。