[發(fā)明專(zhuān)利]關(guān)鍵點(diǎn)檢測(cè)方法、系統(tǒng)、設(shè)備及存儲(chǔ)介質(zhì)

在審

• 申請(qǐng)?zhí)枺?/td>: 201811558885.1
• 申請(qǐng)日: 2018-12-19
• 公開(kāi)（公告）號(hào): CN111340043A
• 公開(kāi)（公告）日: 2020-06-26
• 發(fā)明（設(shè)計(jì)）人: 李翌昕;葉韻
• 申請(qǐng)（專(zhuān)利權(quán)）人: 北京京東尚科信息技術(shù)有限公司;北京京東世紀(jì)貿(mào)易有限公司
• 主分類(lèi)號(hào): G06K9/46
• 分類(lèi)號(hào): G06K9/46;G06K9/34;G06K9/62
• 代理公司: 北京律智知識(shí)產(chǎn)權(quán)代理有限公司 11438
• 代理人: 袁禮君;闞梓瑄

1.一種關(guān)鍵點(diǎn)檢測(cè)方法，其特征在于，包括如下步驟：

將待檢測(cè)圖像輸入訓(xùn)練好的激活圖生成模型，所述激活圖生成模型包括至少一個(gè)階段的激活圖生成模塊，得到至少一個(gè)階段的梯度類(lèi)別激活圖；

將待檢測(cè)圖像和梯度類(lèi)別激活圖輸入訓(xùn)練好的關(guān)鍵點(diǎn)檢測(cè)模型，所述關(guān)鍵點(diǎn)檢測(cè)模型包括與所述激活圖生成模塊一一對(duì)應(yīng)的非局部模塊，所述梯度類(lèi)別激活圖分別輸入對(duì)應(yīng)的非局部模塊，得到關(guān)鍵點(diǎn)預(yù)測(cè)圖；

分析所述關(guān)鍵點(diǎn)預(yù)測(cè)圖中所有連通區(qū)域，計(jì)算面積最大的連通區(qū)域的中心，作為對(duì)應(yīng)關(guān)鍵點(diǎn)的位置。

2.根據(jù)權(quán)利要求1所述的關(guān)鍵點(diǎn)檢測(cè)方法，其特征在于，所述將待檢測(cè)圖像輸入訓(xùn)練好的第一識(shí)別模型之前，還包括如下步驟：

將待檢測(cè)圖像輸入訓(xùn)練好的分類(lèi)模型，得到待檢測(cè)圖像的分類(lèi)結(jié)果；

將待檢測(cè)圖像和分類(lèi)結(jié)果一起輸入訓(xùn)練好的激活圖生成模型。

3.根據(jù)權(quán)利要求1所述的關(guān)鍵點(diǎn)檢測(cè)方法，其特征在于，所述激活圖生成模型中，第一個(gè)激活圖生成模塊之前以及每?jī)蓚€(gè)激活圖生成模塊之間分別設(shè)置有一卷積單元。

4.根據(jù)權(quán)利要求1所述的關(guān)鍵點(diǎn)檢測(cè)方法，其特征在于，所述關(guān)鍵點(diǎn)檢測(cè)模型中，第一非局部模塊之前以及每?jī)蓚€(gè)非局部模塊之間分別設(shè)置有一卷積單元。

5.根據(jù)權(quán)利要求1所述的關(guān)鍵點(diǎn)檢測(cè)方法，其特征在于，所述關(guān)鍵點(diǎn)檢測(cè)模型采用特征金字塔結(jié)構(gòu)。

6.根據(jù)權(quán)利要求1所述的關(guān)鍵點(diǎn)檢測(cè)方法，其特征在于，所述得到關(guān)鍵點(diǎn)預(yù)測(cè)圖之后，還包括如下步驟：

將關(guān)鍵點(diǎn)檢測(cè)模型輸出的關(guān)鍵點(diǎn)預(yù)測(cè)圖輸入優(yōu)化識(shí)別模型，所述優(yōu)化識(shí)別模型包括至少一個(gè)卷積單元；

分析所述優(yōu)化識(shí)別模型輸出的關(guān)鍵點(diǎn)預(yù)測(cè)圖中所有連通區(qū)域，計(jì)算面積最大的連通區(qū)域的中心，作為對(duì)應(yīng)關(guān)鍵點(diǎn)的位置。

7.根據(jù)權(quán)利要求1所述的關(guān)鍵點(diǎn)檢測(cè)方法，其特征在于，所述方法還包括如下步驟：

采集訓(xùn)練用圖像，獲取訓(xùn)練用圖像的關(guān)鍵點(diǎn)位置標(biāo)簽，將訓(xùn)練用圖像和關(guān)鍵點(diǎn)位置標(biāo)簽加入激活圖生成模型的訓(xùn)練集；

采用所述激活圖生成模型的訓(xùn)練集訓(xùn)練所述激活圖生成模型，得到訓(xùn)練好的激活圖生成模型和訓(xùn)練用圖像的梯度類(lèi)別激活圖；

將待檢測(cè)圖像、梯度類(lèi)別激活圖和關(guān)鍵點(diǎn)位置標(biāo)簽加入關(guān)鍵點(diǎn)檢測(cè)模型的訓(xùn)練集；

采用所述關(guān)鍵點(diǎn)檢測(cè)模型的訓(xùn)練集訓(xùn)練所述關(guān)鍵點(diǎn)檢測(cè)模型，得到訓(xùn)練好的關(guān)鍵點(diǎn)檢測(cè)模型。

8.根據(jù)權(quán)利要求7所述的關(guān)鍵點(diǎn)檢測(cè)方法，其特征在于，所述采用所述激活圖生成模型的訓(xùn)練集訓(xùn)練所述激活圖生成模型之前，還包括如下步驟：

獲取訓(xùn)練用圖像的類(lèi)別標(biāo)簽，將訓(xùn)練用圖像和類(lèi)別標(biāo)簽加入分類(lèi)模型的訓(xùn)練集；

采用分類(lèi)模型的訓(xùn)練集訓(xùn)練所述分類(lèi)模型，得到訓(xùn)練好的分類(lèi)模型。

9.根據(jù)權(quán)利要求7所述的關(guān)鍵點(diǎn)檢測(cè)方法，其特征在于，所述得到訓(xùn)練好的關(guān)鍵點(diǎn)檢測(cè)模型之后，還包括如下步驟：

得到關(guān)鍵點(diǎn)檢測(cè)模型輸出的訓(xùn)練用圖像的關(guān)鍵點(diǎn)預(yù)測(cè)圖，將關(guān)鍵點(diǎn)預(yù)測(cè)圖和關(guān)鍵點(diǎn)位置標(biāo)簽加入優(yōu)化識(shí)別模型的訓(xùn)練集，所述優(yōu)化識(shí)別模型包括至少一個(gè)卷積單元，得到訓(xùn)練好的優(yōu)化識(shí)別模型。

10.一種關(guān)鍵點(diǎn)檢測(cè)系統(tǒng)，其特征在于，所述關(guān)鍵點(diǎn)檢測(cè)系統(tǒng)包括：

激活圖生成模型執(zhí)行模塊，用于將待檢測(cè)圖像輸入訓(xùn)練好的激活圖生成模型，所述激活圖生成模型包括至少一個(gè)階段的激活圖生成模塊，得到至少一個(gè)階段的梯度類(lèi)別激活圖；

關(guān)鍵點(diǎn)檢測(cè)模型執(zhí)行模塊，用于將待檢測(cè)圖像和梯度類(lèi)別激活圖輸入訓(xùn)練好的關(guān)鍵點(diǎn)檢測(cè)模型，所述關(guān)鍵點(diǎn)檢測(cè)模型包括與所述激活圖生成模塊一一對(duì)應(yīng)的非局部模塊，所述梯度類(lèi)別激活圖分別輸入對(duì)應(yīng)的非局部模塊，得到關(guān)鍵點(diǎn)預(yù)測(cè)圖；

關(guān)鍵點(diǎn)位置確定模塊，用于分析所述關(guān)鍵點(diǎn)預(yù)測(cè)圖中所有連通區(qū)域，計(jì)算面積最大的連通區(qū)域的中心，作為對(duì)應(yīng)關(guān)鍵點(diǎn)的位置。