[發明專利]關鍵點標注方法、裝置、電子設備及存儲介質

有效

• 申請號: 202010530847.6
• 申請日: 2020-06-11
• 公開（公告）號: CN111695628B
• 公開（公告）日: 2023-05-05
• 發明（設計）人: 宮延河;彭昊天
• 申請（專利權）人: 北京百度網訊科技有限公司
• 主分類號: G06V20/64
• 分類號: G06V20/64;G06V10/75;G06V10/82;G06T17/00;G06T15/00
• 代理公司: 北京銀龍知識產權代理有限公司 11243
• 代理人: 曹娜

本申請公開了關鍵點標注方法、裝置、電子設備及存儲介質，涉及人工智能及深度學習領域。具體實現方案為：獲取包含目標物體的待標注圖像；從預先保存的多個三維模型中，獲取與該目標物體的匹配度最高的第一三維模型；將該第一三維模型覆蓋在該待標注圖像上，并旋轉該第一三維模型；在該第一三維模型的渲染圖像與該待標注圖像中的目標物體的重合面積最大時，獲取該第一三維模型的旋轉角度；根據該第一三維模型上的參考關鍵點及該旋轉角度，確定參考關鍵點在該待標注圖像中的坐標。本申請實施例能夠節約關鍵點標注的時間和人力成本，并且能夠提升關鍵點標注的精度和可靠性。

技術領域

本申請涉及計算機技術領域，尤其涉及人工智能及深度學習領域。

背景技術

深度學習是利用計算機的數據處理優勢，利用大的數據訓練某一類型的識別模型，最終使識別模型獲取高效的數據特征識別能力和分類能力。針對不同目的的識別模型進行數據訓練，需要大量與識別目的的相關的數據集。現有技術中經常采用有監督學習提高識別模型的學習效率和識別效果。有監督模型需要識別對象的特征樣本，即需要在數據集中提供被識別對象的標注樣本數據，標注樣本數據的準確度關系到深度學習的最終效果。目前大部分的樣本數據采用人工標注，例如，為了完成鞋子的關鍵點檢測，通常由人工進行鞋子的關鍵點標注。由于鞋子存在嚴重的遮擋問題，在旋轉一定角度后鞋子的一多半部分均處于不可見狀態；這種情況導致人工標注的關鍵點會出現較大的偏差，限制了后續模型訓練精度的進一步提升。

發明內容

本申請提供了一種用于關鍵點標注方法、裝置、電子設備及存儲介質。

根據本申請的一方面，提供了一種關鍵點標注方法，包括：

獲取包含目標物體的待標注圖像；

從預先保存的多個三維模型中，獲取與所述目標物體的匹配度最高的第一三維模型；

將所述第一三維模型覆蓋在所述待標注圖像上，并旋轉所述第一三維模型；在所述第一三維模型的渲染圖像與所述待標注圖像中的目標物體的重合面積最大時，獲取所述第一三維模型的旋轉角度；

根據所述第一三維模型上的參考關鍵點及所述旋轉角度，確定所述參考關鍵點在所述待標注圖像中的坐標。

根據本申請的另一方面，提供了一種關鍵點標注裝置，包括：

待標注圖像獲取模塊，用于獲取包含目標物體的待標注圖像；

三維模型獲取模塊，用于從預先保存的多個三維模型中，獲取與所述目標物體的匹配度最高的第一三維模型；

調整模塊，用于將所述第一三維模型覆蓋在所述待標注圖像上，并旋轉所述第一三維模型；在所述第一三維模型的渲染圖像與所述待標注圖像中的目標物體的重合面積最大時，獲取所述第一三維模型的旋轉角度；

標注模塊，用于根據所述第一三維模型上的參考關鍵點及所述旋轉角度，確定所述參考關鍵點在所述待標注圖像中的坐標。

根據本申請的另一方面，提供了一種電子設備，包括：

至少一個處理器；以及

與所述至少一個處理器通信連接的存儲器；其中，

所述存儲器存儲有可被所述至少一個處理器執行的指令，所述指令被所述至少一個處理器執行，以使所述至少一個處理器能夠執行關鍵點標注方法中任一項所述的方法。

根據本申請的另一方面，提供了一種存儲有計算機指令的非瞬時計算機可讀存儲介質，所述計算機指令用于使所述計算機執行關鍵點標注方法中任一項所述的方法。

根據本申請的另一方面，提供了一種計算機程序產品，包括計算機程序，所述計算機程序在被處理器執行時實現如上所述的方法。