[發(fā)明專利]鋼筋檢測模型的訓練方法及裝置

在審

• 申請?zhí)枺?/td>: 202211642914.9
• 申請日: 2022-12-20
• 公開（公告）號: CN115965850A
• 公開（公告）日: 2023-04-14
• 發(fā)明（設計）人: 請求不公布姓名
• 申請（專利權）人: 深圳須彌云圖空間科技有限公司
• 主分類號: G06V10/82
• 分類號: G06V10/82;G06V10/764;G06N3/0464;G06N3/08
• 代理公司: 北京嘉科知識產(chǎn)權代理事務所(特殊普通合伙) 11687
• 代理人: 陳美君

1.一種鋼筋檢測模型的訓練方法，其特征在于，包括：

將RetinaNet模型中的普通卷積層替換為空洞卷積層，得到鋼筋檢測模型；

獲取鋼筋訓練數(shù)據(jù)集，將所述鋼筋訓練數(shù)據(jù)集中的訓練樣本輸入所述鋼筋檢測模型，輸出所述訓練樣本的特征圖，其中，所述特征圖包含多個中心點；

利用平衡度公式計算每個中心點的平衡度，基于每個中心點的平衡度對該中心點的進行分類，并在分類的過程中利用第一損失函數(shù)訓練所述鋼筋檢測模型，其中，每個中心點的類別包括前景和背景；

基于所述訓練樣本中的每個標簽對應的被分類為前景的一個或多個中心點回歸該標簽對應的檢測框的坐標，其中，所述訓練樣本中包含多個標簽，并在回歸的過程中利用第二損失函數(shù)訓練所述鋼筋檢測模型；

基于所述訓練樣本中所有檢測框中每兩個相鄰檢測框的交并比，利用第三損失函數(shù)訓練所述鋼筋檢測模型。

2.根據(jù)權利要求1所述的方法，其特征在于，利用平衡度公式計算每個中心點的平衡度，包括：

分別計算每個中心點到該中心點對應標簽的上邊框、下邊框、左邊框和右邊框的距離；

基于每個中心點到該中心點對應標簽的上邊框、下邊框、左邊框和右邊框的距離，利用所述平衡度公式計算每個中心點的所述平衡度。

3.根據(jù)權利要求1所述的方法，其特征在于，利用平衡度公式計算每個中心點的平衡度p：

其中，每個中心點到該中心點對應標簽的左邊框的距離為l，每個中心點到該中心點對應標簽的右邊框的距離為r，每個中心點到該中心點對應標簽的上邊框的距離為t，每個中心點到該中心點對應標簽的下邊框的距離為b。

4.根據(jù)權利要求1所述的方法，其特征在于，基于每個中心點的平衡度對該中心點的進行分類，包括：

將所述平衡度大于預設閾值的中心點分類為前景；

將所述平衡度小于等于所述預設閾值的中心點分類為背景。

5.根據(jù)權利要求1所述的方法，其特征在于，基于所述訓練樣本中的每個標簽對應的被分類為前景的一個或多個中心點回歸該標簽對應的檢測框的坐標，包括：

將處于每個標簽內(nèi)的被分類為前景的中心點確定為該標簽對應的中心點；

基于每個標簽的框高和框?qū)捯约霸摌撕瀸囊粋€或多個中心點的坐標回歸該標簽對應的檢測框的坐標。

6.根據(jù)權利要求1所述的方法，其特征在于，基于所述訓練樣本中的每個標簽對應的被分類為前景的一個或多個中心點回歸該標簽對應的檢測框的坐標，包括：

將處于每個標簽內(nèi)的被分類為前景的中心點以及處于每個標簽內(nèi)的被分類為前景的中心點在上、下、左、右四方向上相鄰的中心點確定為該標簽對應的中心點；

基于每個標簽的框高和框?qū)捯约霸摌撕瀸囊粋€或多個中心點的坐標回歸該標簽對應的檢測框的坐標。

7.根據(jù)權利要求1所述的方法，其特征在于，第三損失函數(shù)：

其中，n為所述訓練樣本中檢測框的總數(shù)，b_i為第i個檢測框，b_j為第j個檢測框，gt_i為第i個檢測框?qū)臉撕灒琺為所述訓練樣本中標簽的總數(shù)，IoU為交并比函數(shù)。