[發(fā)明專利]位姿識(shí)別模型訓(xùn)練方法、裝置、位姿識(shí)別方法和終端設(shè)備

在審

• 申請(qǐng)?zhí)枺?/td>: 202011503576.1
• 申請(qǐng)日: 2020-12-18
• 公開（公告）號(hào): CN112489129A
• 公開（公告）日: 2021-03-12
• 發(fā)明（設(shè)計(jì)）人: 林燦然;郭渺辰;程駿;龐建新
• 申請(qǐng)（專利權(quán)）人: 深圳市優(yōu)必選科技股份有限公司
• 主分類號(hào): G06T7/73
• 分類號(hào): G06T7/73;G06T15/00;G06K9/62;G06N3/04;G06N3/08
• 代理公司: 北京超凡宏宇專利代理事務(wù)所(特殊普通合伙) 11463
• 代理人: 梁韜

1.一種位姿識(shí)別模型訓(xùn)練方法，其特征在于，該位姿識(shí)別模型訓(xùn)練方法包括：

獲取原始RGB圖像；

利用預(yù)設(shè)的位姿點(diǎn)線圖提取方法從所述原始RGB圖像中提取由預(yù)設(shè)數(shù)目個(gè)位姿關(guān)鍵點(diǎn)坐標(biāo)組成的位姿點(diǎn)線圖；

將所述位姿點(diǎn)線圖以區(qū)別于預(yù)定背景板顏色的目標(biāo)顏色繪制在背景板上，以確定訓(xùn)練RGB圖像；

利用多個(gè)訓(xùn)練RGB圖像訓(xùn)練所述位姿識(shí)別模型直至所述位姿識(shí)別模型對(duì)應(yīng)的損失函數(shù)收斂。

2.根據(jù)權(quán)利要求1所述的位姿識(shí)別模型訓(xùn)練方法，其特征在于，在從所述原始RGB圖像中提取由預(yù)設(shè)數(shù)目個(gè)位姿關(guān)鍵點(diǎn)坐標(biāo)組成的位姿點(diǎn)線圖之后，還包括：

根據(jù)所述位姿點(diǎn)線圖中各個(gè)位姿關(guān)鍵點(diǎn)坐標(biāo)的屬性對(duì)各個(gè)位姿關(guān)鍵點(diǎn)坐標(biāo)進(jìn)行特征分類。

3.根據(jù)權(quán)利要求2所述的位姿識(shí)別模型訓(xùn)練方法，其特征在于，所述將所述位姿點(diǎn)線圖以區(qū)別于預(yù)定背景板顏色的目標(biāo)顏色分別繪制在背景板上，包括：

將所述位姿點(diǎn)線圖繪制在所述背景板上；

在所述背景板上，將同一特征類別中的位姿關(guān)鍵點(diǎn)坐標(biāo)構(gòu)成的點(diǎn)線標(biāo)注為預(yù)設(shè)的同一顏色，各個(gè)特征類別中的位姿關(guān)鍵點(diǎn)坐標(biāo)構(gòu)成的點(diǎn)線標(biāo)注顏色不同且與所述預(yù)定背景板顏色不同。

4.根據(jù)權(quán)利要求3所述的位姿識(shí)別模型訓(xùn)練方法，其特征在于，在識(shí)別人體位姿時(shí)，所述特征類別包括頭頸特征類別、手臂特征類別和下肢特征類別。

5.根據(jù)權(quán)利要求4所述的位姿識(shí)別模型訓(xùn)練方法，其特征在于，所述預(yù)定背景板顏色為黑色，所述頭頸特征類別、所述手臂特征類別和所述下肢特征類別中的位姿關(guān)鍵點(diǎn)坐標(biāo)構(gòu)成的點(diǎn)線標(biāo)注為綠色、藍(lán)色和紅色之一。

6.根據(jù)權(quán)利要求1至5任一項(xiàng)所述的位姿識(shí)別模型訓(xùn)練方法，其特征在于，所述利用預(yù)設(shè)的位姿點(diǎn)線圖提取方法從所述原始RGB圖像中提取由預(yù)設(shè)數(shù)目個(gè)位姿關(guān)鍵點(diǎn)坐標(biāo)組成的位姿點(diǎn)線圖，包括：

利用OpenPose人體姿態(tài)估計(jì)算法提取所述原始RGB圖像的預(yù)設(shè)數(shù)目個(gè)關(guān)鍵點(diǎn)的位置信息；

根據(jù)預(yù)設(shè)數(shù)目個(gè)關(guān)鍵點(diǎn)的位置信息確定所述位姿點(diǎn)線圖。

7.一種位姿識(shí)別模型訓(xùn)練裝置，其特征在于，該裝置包括：

原始RGB圖像獲取模塊，用于獲取原始RGB圖像；

位姿點(diǎn)線圖提取模塊，用于利用預(yù)設(shè)的位姿點(diǎn)線圖提取方法從所述原始RGB圖像中提取由預(yù)設(shè)數(shù)目個(gè)位姿關(guān)鍵點(diǎn)坐標(biāo)組成的位姿點(diǎn)線圖；

訓(xùn)練RGB圖像確定模塊，用于將所述位姿點(diǎn)線圖以區(qū)別于預(yù)定背景板顏色的目標(biāo)顏色繪制在背景板上，以確定訓(xùn)練RGB圖像；

位姿識(shí)別模型訓(xùn)練模塊，用于利用多個(gè)訓(xùn)練RGB圖像訓(xùn)練所述位姿識(shí)別模型直至所述位姿識(shí)別模型對(duì)應(yīng)的損失函數(shù)收斂。

8.一種位姿識(shí)別方法，其特征在于，該位姿識(shí)別方法包括：

獲取待識(shí)別的初始RGB圖像；

利用預(yù)設(shè)的位姿點(diǎn)線圖提取方法從所述初始RGB圖像中提取由預(yù)設(shè)數(shù)目個(gè)位姿關(guān)鍵點(diǎn)坐標(biāo)組成的位姿點(diǎn)線圖；

將所述位姿點(diǎn)線圖以區(qū)別于所述預(yù)定背景板顏色的目標(biāo)顏色分別繪制在所述背景板上，以確定待識(shí)別的點(diǎn)線RGB圖像；

利用權(quán)利要求1至5任一項(xiàng)所述位姿識(shí)別模型訓(xùn)練方法訓(xùn)練獲得的位姿識(shí)別模型確定所述點(diǎn)線RGB圖像對(duì)應(yīng)的位姿。

9.一種終端設(shè)備，其特征在于，包括存儲(chǔ)器和處理器，所述存儲(chǔ)器用于存儲(chǔ)計(jì)算機(jī)程序，所述計(jì)算機(jī)程序在所述處理器上運(yùn)行時(shí)執(zhí)行權(quán)利要求1至6任一項(xiàng)所述的位姿識(shí)別模型訓(xùn)練方法或權(quán)利要求8所述的位姿識(shí)別方法。

10.一種可讀存儲(chǔ)介質(zhì)，其特征在于，其存儲(chǔ)有計(jì)算機(jī)程序，所述計(jì)算機(jī)程序在處理器上運(yùn)行時(shí)執(zhí)行權(quán)利要求1至6任一項(xiàng)所述的位姿識(shí)別模型訓(xùn)練方法或權(quán)利要求8所述的位姿識(shí)別方法。