[發(fā)明專利]模型優(yōu)化方法、裝置、計(jì)算機(jī)設(shè)備和存儲(chǔ)介質(zhì)

在審

• 申請(qǐng)?zhí)枺?/td>: 202110349990.X
• 申請(qǐng)日: 2021-03-31
• 公開(kāi)（公告）號(hào): CN113705308A
• 公開(kāi)（公告）日: 2021-11-26
• 發(fā)明（設(shè)計(jì)）人: 彭健騰
• 申請(qǐng)（專利權(quán)）人: 騰訊科技（深圳）有限公司
• 主分類號(hào): G06K9/00
• 分類號(hào): G06K9/00;G06K9/46;G06K9/62
• 代理公司: 廣州三環(huán)專利商標(biāo)代理有限公司 44202
• 代理人: 熊永強(qiáng);杜維

1.一種模型優(yōu)化方法，其特征在于，包括：

獲取樣本圖像，所述樣本圖像包括樣本對(duì)象；

調(diào)用對(duì)象檢測(cè)模型對(duì)所述樣本圖像進(jìn)行對(duì)象檢測(cè)，得到所述樣本對(duì)象的目標(biāo)關(guān)鍵點(diǎn)的位置檢測(cè)信息，及所述目標(biāo)關(guān)鍵點(diǎn)的參考對(duì)象的位置檢測(cè)信息；

獲取所述對(duì)象檢測(cè)模型的位置損失函數(shù)，所述位置損失函數(shù)是根據(jù)所述目標(biāo)關(guān)鍵點(diǎn)和所述參考對(duì)象之間需滿足的目標(biāo)位置關(guān)系構(gòu)建的；

采用所述位置損失函數(shù)，根據(jù)所述目標(biāo)關(guān)鍵點(diǎn)的位置檢測(cè)信息與所述參考對(duì)象的位置檢測(cè)信息，計(jì)算目標(biāo)損失值；

根據(jù)所述目標(biāo)損失值優(yōu)化所述對(duì)象檢測(cè)模型的模型參數(shù)，得到優(yōu)化后的對(duì)象檢測(cè)模型。

2.如權(quán)利要求1所述的方法，其特征在于，所述樣本對(duì)象包括一個(gè)或者多個(gè)關(guān)鍵點(diǎn)；所述參考對(duì)象包括所述樣本對(duì)象的對(duì)象檢測(cè)框；

所述位置損失函數(shù)包括第一損失函數(shù)，所述目標(biāo)損失值包括：所述第一損失函數(shù)的第一損失值；其中，所述目標(biāo)位置關(guān)系包括：所述樣本對(duì)象的任一關(guān)鍵點(diǎn)和所述對(duì)象檢測(cè)框之間需滿足的第一位置關(guān)系，所述第一損失函數(shù)是基于所述第一位置關(guān)系構(gòu)建的，所述第一位置關(guān)系用于指示所述任一關(guān)鍵點(diǎn)需位于所述對(duì)象檢測(cè)框內(nèi)。

3.如權(quán)利要求2所述的方法，其特征在于，所述采用所述位置損失函數(shù)，根據(jù)所述目標(biāo)關(guān)鍵點(diǎn)的位置檢測(cè)信息與所述參考對(duì)象的位置檢測(cè)信息，計(jì)算目標(biāo)損失值，包括：

采用所述第一損失函數(shù)，根據(jù)所述目標(biāo)關(guān)鍵點(diǎn)的位置檢測(cè)信息與所述參考對(duì)象的位置檢測(cè)信息，計(jì)算所述目標(biāo)關(guān)鍵點(diǎn)對(duì)應(yīng)的損失值；

根據(jù)所述目標(biāo)關(guān)鍵點(diǎn)對(duì)應(yīng)的損失值，確定第一損失值。

4.如權(quán)利要求3所述的方法，其特征在于，所述目標(biāo)關(guān)鍵點(diǎn)的位置檢測(cè)信息包括：所述目標(biāo)關(guān)鍵點(diǎn)在所述樣本圖像中的位置坐標(biāo)，且所述位置坐標(biāo)包括橫坐標(biāo)值和縱坐標(biāo)值；

所述對(duì)象檢測(cè)框?yàn)榫匦芜吔缈颍鰠⒖紝?duì)象的位置檢測(cè)信息包括：所述對(duì)象檢測(cè)框的各條邊界在所述樣本圖像中的位置指示值；其中，所述第一位置關(guān)系通過(guò)所述任一關(guān)鍵點(diǎn)和每條邊界之間需滿足的方位關(guān)系進(jìn)行表示；

所述采用所述第一損失函數(shù)，根據(jù)所述目標(biāo)關(guān)鍵點(diǎn)的位置檢測(cè)信息與所述參考對(duì)象的位置檢測(cè)信息，計(jì)算所述目標(biāo)關(guān)鍵點(diǎn)對(duì)應(yīng)的損失值，包括：

遍歷所述對(duì)象檢測(cè)框的各條邊界；

從所述參考對(duì)象的位置檢測(cè)信息中獲取當(dāng)前遍歷的當(dāng)前邊界的位置指示值，并從所述目標(biāo)關(guān)鍵點(diǎn)的位置坐標(biāo)中，獲取與所述當(dāng)前邊界相關(guān)的坐標(biāo)值；

采用所述第一損失函數(shù)，根據(jù)所述當(dāng)前邊界的位置指示值和獲取到的坐標(biāo)值，計(jì)算所述當(dāng)前邊界對(duì)應(yīng)的損失值；

在所述各條邊界均被遍歷后，對(duì)所述各條邊界對(duì)應(yīng)的損失值進(jìn)行求和運(yùn)算，得到所述目標(biāo)關(guān)鍵點(diǎn)對(duì)應(yīng)的損失值。

5.如權(quán)利要求3所述的方法，其特征在于，所述目標(biāo)關(guān)鍵點(diǎn)的位置檢測(cè)信息包括：所述目標(biāo)關(guān)鍵點(diǎn)在所述樣本圖像中的位置坐標(biāo)；

所述對(duì)象檢測(cè)框?yàn)椴灰?guī)則檢測(cè)框，所述參考對(duì)象的位置檢測(cè)信息包括：所述樣本檢測(cè)框的各個(gè)頂點(diǎn)在所述樣本圖像中的位置坐標(biāo)；所述對(duì)象檢測(cè)框采用多個(gè)頂點(diǎn)向量表示，任一頂點(diǎn)向量是根據(jù)所述對(duì)象檢測(cè)框的兩個(gè)相鄰頂點(diǎn)的位置坐標(biāo)確定的；其中，所述第一位置關(guān)系通過(guò)所述任一關(guān)鍵點(diǎn)和每個(gè)頂點(diǎn)向量之間需滿足的方位關(guān)系進(jìn)行表示；

所述采用所述第一損失函數(shù)，根據(jù)所述目標(biāo)關(guān)鍵點(diǎn)的位置檢測(cè)信息與所述參考對(duì)象的位置檢測(cè)信息，計(jì)算所述目標(biāo)關(guān)鍵點(diǎn)對(duì)應(yīng)的損失值，包括：

遍歷所述對(duì)象檢測(cè)框的多個(gè)頂點(diǎn)，并確定當(dāng)前遍歷的當(dāng)前頂點(diǎn)所對(duì)應(yīng)的目標(biāo)頂點(diǎn)向量，所述目標(biāo)頂點(diǎn)向量是指采用所述當(dāng)前頂點(diǎn)作為起始點(diǎn)所構(gòu)建的頂點(diǎn)向量；

從所述參考對(duì)象的位置信息中獲取所述當(dāng)前頂點(diǎn)的位置坐標(biāo)，并根據(jù)所述當(dāng)前頂點(diǎn)的位置坐標(biāo)和所述目標(biāo)關(guān)鍵點(diǎn)的位置坐標(biāo)，計(jì)算所述當(dāng)前頂點(diǎn)和所述目標(biāo)關(guān)鍵點(diǎn)之間的參考頂點(diǎn)向量；

采用所述第一損失函數(shù)，根據(jù)所述參考頂點(diǎn)向量和所述目標(biāo)頂點(diǎn)向量，計(jì)算所述當(dāng)前頂點(diǎn)對(duì)應(yīng)的損失值；

在所述對(duì)象檢測(cè)框的各個(gè)頂點(diǎn)均被遍歷后，對(duì)所述各個(gè)頂點(diǎn)對(duì)應(yīng)的損失值進(jìn)行求和運(yùn)算，得到所述目標(biāo)關(guān)鍵點(diǎn)對(duì)應(yīng)的損失值。