[發(fā)明專(zhuān)利]用于翡翠圖像分割的方法及模型訓(xùn)練方法

有效

• 申請(qǐng)?zhí)枺?/td>: 202011353434.1
• 申請(qǐng)日: 2020-11-26
• 公開(kāi)（公告）號(hào): CN112465835B
• 公開(kāi)（公告）日: 2022-07-08
• 發(fā)明（設(shè)計(jì)）人: 易金鵬;王秀輝;劉繼澎
• 申請(qǐng)（專(zhuān)利權(quán)）人: 深圳市對(duì)莊科技有限公司
• 主分類(lèi)號(hào): G06T7/11
• 分類(lèi)號(hào): G06T7/11;G06T7/13;G06T7/136;G06N3/04;G06N3/08;G06N20/00;G06T5/40;G06T5/50
• 代理公司: 深圳市精英專(zhuān)利事務(wù)所 44242
• 代理人: 譚穗平

本公開(kāi)實(shí)施例公開(kāi)了一種用于翡翠圖像分割的方法及模型訓(xùn)練方法，其中用于翡翠圖像分割的方法包括：首先響應(yīng)于獲取到包含翡翠圖像的原始圖像，將所述原始圖像輸入至基于機(jī)器學(xué)習(xí)預(yù)訓(xùn)練的第一網(wǎng)絡(luò)模型中，以使第一網(wǎng)絡(luò)模型對(duì)原始圖像進(jìn)行分割，輸出包含無(wú)背景圖的初始翡翠圖像；而后將初始翡翠圖像輸入至基于機(jī)器學(xué)習(xí)預(yù)訓(xùn)練的第二網(wǎng)絡(luò)模型中，以使第二網(wǎng)絡(luò)模型對(duì)初始翡翠圖像的高光區(qū)域圖像進(jìn)行分割，輸出不包含高光區(qū)域圖像的翡翠圖像。采用機(jī)器學(xué)習(xí)訓(xùn)練模型能夠準(zhǔn)確地去除翡翠圖像的背景和高光區(qū)域進(jìn)而解決了翡翠圖像分割采用數(shù)字圖像處理，適用場(chǎng)景單一、分割邊界模糊以及分割不準(zhǔn)確的技術(shù)問(wèn)題。

技術(shù)領(lǐng)域

本公開(kāi)涉及圖像分割處理技術(shù)領(lǐng)域，具體涉及到一種用于翡翠圖像分割的方法及模型訓(xùn)練方法。

背景技術(shù)

翡翠圖像分割大多使用傳統(tǒng)數(shù)字圖像處理，一方面適用場(chǎng)景單一，且分割效率低；另一方面存在分割邊界模糊，分割不準(zhǔn)。

發(fā)明內(nèi)容

本公開(kāi)的主要目的在于提供一種用于翡翠圖像分割的方法及模型訓(xùn)練方法，以解決傳統(tǒng)的數(shù)字圖像處理技術(shù)適用場(chǎng)景單一，且分割不準(zhǔn)確的問(wèn)題。

為了實(shí)現(xiàn)上述目的，根據(jù)本公開(kāi)的第一方面，提供了一種用于翡翠圖像分割的方法，包括：響應(yīng)于獲取到包含翡翠圖像的原始圖像，將所述原始圖像輸入至基于機(jī)器學(xué)習(xí)預(yù)訓(xùn)練的第一網(wǎng)絡(luò)模型中，以使所述第一網(wǎng)絡(luò)模型對(duì)所述原始圖像進(jìn)行分割，輸出包含無(wú)背景圖的初始翡翠圖像；將所述初始翡翠圖像輸入至基于機(jī)器學(xué)習(xí)預(yù)訓(xùn)練的第二網(wǎng)絡(luò)模型中，以使所述第二網(wǎng)絡(luò)模型對(duì)所述初始翡翠圖像的高光區(qū)域圖像進(jìn)行分割，輸出不包含高光區(qū)域圖像的翡翠圖像。

可選地，響應(yīng)于獲取到原始圖像，將所述原始圖像輸入至基于機(jī)器學(xué)習(xí)預(yù)訓(xùn)練的第一網(wǎng)絡(luò)模型中，以使所述第一網(wǎng)絡(luò)模型對(duì)所述原始圖像進(jìn)行分割，輸出包含無(wú)背景圖的初始翡翠圖像包括：響應(yīng)于獲取到視頻中每一幀的圖像，確定視頻中所有幀的圖像中滿(mǎn)足預(yù)設(shè)條件的圖像；將所述滿(mǎn)足預(yù)設(shè)條件的圖像輸入至基于機(jī)器學(xué)習(xí)預(yù)訓(xùn)練的第一網(wǎng)絡(luò)模型中，以使所述第一網(wǎng)絡(luò)模型對(duì)所述原始圖像進(jìn)行分割，輸出包含無(wú)背景圖的初始翡翠圖像。

根據(jù)本公開(kāi)的第二方面，提供了一種用于翡翠圖像分割的模型訓(xùn)練方法，包括對(duì)原始圖像集中每一張包含翡翠的原始圖像預(yù)處理，得到原始圖像中翡翠圖像的掩膜；將作為正樣本的所述掩膜和作為負(fù)樣本的所述原始圖像集中不包含翡翠的原始圖像按照預(yù)設(shè)的數(shù)量比輸入至語(yǔ)義分割網(wǎng)絡(luò)模型中，以訓(xùn)練所述語(yǔ)義分割網(wǎng)絡(luò)模型識(shí)別圖像中的背景區(qū)域圖像和翡翠圖像，并將背景區(qū)域圖像和翡翠圖像分割。

根據(jù)本公開(kāi)的第三方面，提供了一種用于翡翠圖像分割的模型訓(xùn)練方法，包括獲取與背景區(qū)域圖像分割的翡翠圖像；基于圖像高光估計(jì)算法，選取所述翡翠圖像的高光區(qū)域圖像；將作為正樣本的所述高光區(qū)域圖像和作為負(fù)樣本的非高光翡翠圖像按照預(yù)設(shè)的數(shù)量比輸入至第一語(yǔ)義分割模型中，以訓(xùn)練所述第一語(yǔ)義分割模型識(shí)別翡翠圖像中的高光區(qū)域圖像和非高光區(qū)域圖像，并將高光區(qū)域圖像和非高光區(qū)域圖像進(jìn)行分割。

根據(jù)本公開(kāi)的第四方面，提供了一種用于翡翠圖像分割的裝置，包括：第一輸入單元，被配置成響應(yīng)于獲取到包含翡翠圖像的原始圖像，將所述原始圖像輸入至基于機(jī)器學(xué)習(xí)預(yù)訓(xùn)練的第一網(wǎng)絡(luò)模型中，以使所述第一網(wǎng)絡(luò)模型對(duì)所述原始圖像進(jìn)行分割，輸出包含無(wú)背景圖的初始翡翠圖像；第二輸入單元，被配置成將所述初始翡翠圖像輸入至基于機(jī)器學(xué)習(xí)預(yù)訓(xùn)練的第二網(wǎng)絡(luò)模型中，以使所述第二網(wǎng)絡(luò)模型對(duì)所述初始翡翠圖像的高光區(qū)域圖像進(jìn)行分割，輸出不包含高光區(qū)域圖像的翡翠圖像。

根據(jù)本公開(kāi)的第五方面，提供了一種計(jì)算機(jī)可讀存儲(chǔ)介質(zhì)，計(jì)算機(jī)可讀存儲(chǔ)介質(zhì)存儲(chǔ)有計(jì)算機(jī)指令，所述計(jì)算機(jī)指令用于使所述計(jì)算機(jī)執(zhí)行第一方面任意一項(xiàng)實(shí)施例所述的用于翡翠圖像分割的方法。

根據(jù)本公開(kāi)的第六方面，提供了一種電子設(shè)備，包括：至少一個(gè)處理器；以及與所述至少一個(gè)處理器通信連接的存儲(chǔ)器；其中，所述存儲(chǔ)器存儲(chǔ)有可被所述至少一個(gè)處理器執(zhí)行的計(jì)算機(jī)程序，所述計(jì)算機(jī)程序被所述至少一個(gè)處理器執(zhí)行，以使所述至少一個(gè)處理器執(zhí)行第一方面任意一項(xiàng)實(shí)施例所述的用于翡翠圖像分割的方法。