[發(fā)明專(zhuān)利]圖像實(shí)例分割優(yōu)化處理方法、裝置和計(jì)算機(jī)設(shè)備

在審

• 申請(qǐng)?zhí)枺?/td>: 202010194266.X
• 申請(qǐng)日: 2020-03-19
• 公開(kāi)（公告）號(hào): CN111507985A
• 公開(kāi)（公告）日: 2020-08-07
• 發(fā)明（設(shè)計(jì)）人: 張少林;寧欣;石園;劉江寬;支金林
• 申請(qǐng)（專(zhuān)利權(quán)）人: 北京市威富安防科技有限公司
• 主分類(lèi)號(hào): G06T7/11
• 分類(lèi)號(hào): G06T7/11;G06N3/04
• 代理公司: 廣州華進(jìn)聯(lián)合專(zhuān)利商標(biāo)代理有限公司 44224
• 代理人: 王寧

1.一種圖像實(shí)例分割優(yōu)化處理方法，其特征在于，所述方法包括：

獲取圖像實(shí)例分割任務(wù)，所述圖像實(shí)例分割任務(wù)攜帶待分割圖像以及模型標(biāo)識(shí)；

對(duì)所述模型標(biāo)識(shí)對(duì)應(yīng)的實(shí)例分割模型進(jìn)行解析，得到所述實(shí)例分割模型對(duì)應(yīng)的原模型網(wǎng)絡(luò)；

調(diào)用預(yù)設(shè)子網(wǎng)絡(luò)，根據(jù)所述預(yù)設(shè)子網(wǎng)絡(luò)對(duì)所述原模型網(wǎng)絡(luò)進(jìn)行更換，得到目標(biāo)模型網(wǎng)絡(luò)；

對(duì)所述目標(biāo)模型網(wǎng)絡(luò)中的所述預(yù)設(shè)子網(wǎng)絡(luò)進(jìn)行剪枝處理，得到剪枝處理后的實(shí)例分割模型；

將所述待分割圖像輸入至所述剪枝處理后的實(shí)例分割模型，對(duì)所述待分割圖像進(jìn)行預(yù)測(cè)運(yùn)算，得到圖像實(shí)例分割結(jié)果。

2.根據(jù)權(quán)利要求1所述的方法，其特征在于，所述對(duì)所述目標(biāo)模型網(wǎng)絡(luò)中的所述預(yù)設(shè)子網(wǎng)絡(luò)進(jìn)行剪枝處理，得到剪枝處理后的實(shí)例分割模型包括：

獲取與所述預(yù)設(shè)子網(wǎng)絡(luò)對(duì)應(yīng)的通道縮放因子；

根據(jù)所述通道縮放因子將所述目標(biāo)模型網(wǎng)絡(luò)中所述預(yù)設(shè)子網(wǎng)絡(luò)的網(wǎng)絡(luò)通道進(jìn)行排列；

對(duì)所述排列后的網(wǎng)絡(luò)通道進(jìn)行剪枝處理，得到剪枝處理后的實(shí)例分割模型。

3.根據(jù)權(quán)利要求1所述的方法，其特征在于，所述根據(jù)所述預(yù)設(shè)子網(wǎng)絡(luò)對(duì)所述原模型網(wǎng)絡(luò)進(jìn)行更換，得到目標(biāo)模型網(wǎng)絡(luò)包括：

根據(jù)所述預(yù)設(shè)子網(wǎng)絡(luò)在所述原模型網(wǎng)絡(luò)中確定待更換子網(wǎng)絡(luò)；

將所述預(yù)設(shè)子網(wǎng)絡(luò)對(duì)所述待更換子網(wǎng)絡(luò)進(jìn)行更換，得到目標(biāo)模型網(wǎng)絡(luò)。

4.根據(jù)權(quán)利要求1所述的方法，其特征在于，在根據(jù)所述預(yù)設(shè)子網(wǎng)絡(luò)對(duì)所述原模型網(wǎng)絡(luò)進(jìn)行更換，得到目標(biāo)模型網(wǎng)絡(luò)之后，還包括：

獲取訓(xùn)練圖像集；

通過(guò)所述訓(xùn)練圖像集對(duì)所述目標(biāo)模型網(wǎng)絡(luò)進(jìn)行訓(xùn)練，直至滿足第一訓(xùn)練條件，得到第一模型網(wǎng)絡(luò)。

5.根據(jù)權(quán)利要求4所述的方法，其特征在于，在所述獲取訓(xùn)練圖像集之后，還包括：

對(duì)所述訓(xùn)練圖像集進(jìn)行預(yù)處理，得到預(yù)處理后的訓(xùn)練圖像集；

通過(guò)第一訓(xùn)練策略以及所述預(yù)處理后的訓(xùn)練圖像集對(duì)所述目標(biāo)模型網(wǎng)絡(luò)進(jìn)行訓(xùn)練。

6.根據(jù)權(quán)利要求1所述的方法，其特征在于，在根據(jù)所述預(yù)設(shè)子網(wǎng)絡(luò)對(duì)所述原模型網(wǎng)絡(luò)進(jìn)行更換，得到目標(biāo)模型網(wǎng)絡(luò)之后，還包括：

對(duì)所述目標(biāo)模型網(wǎng)絡(luò)進(jìn)行正則化處理，得到正則化處理后的目標(biāo)模型網(wǎng)絡(luò)；

獲取訓(xùn)練圖像集；

通過(guò)第二訓(xùn)練策略以及所述訓(xùn)練圖像集對(duì)所述正則化處理后的目標(biāo)模型網(wǎng)絡(luò)進(jìn)行稀疏化訓(xùn)練，直至滿足第二訓(xùn)練條件，得到第二模型網(wǎng)絡(luò)。

7.根據(jù)權(quán)利要求1至6所述的方法，其特征在于，對(duì)所述目標(biāo)模型網(wǎng)絡(luò)中的所述預(yù)設(shè)子網(wǎng)絡(luò)進(jìn)行剪枝處理，得到剪枝處理后的實(shí)例分割模型之后，還包括：

將所述剪枝處理后的實(shí)例分割模型中的正則化部分進(jìn)行過(guò)濾；

獲取訓(xùn)練圖像集；

根據(jù)所述訓(xùn)練圖像集對(duì)所述過(guò)濾后的實(shí)例分割模型進(jìn)行調(diào)整，直至滿足預(yù)設(shè)條件。

8.一種圖像分割優(yōu)化處理裝置，其特征在于，所述裝置包括：

獲取模塊，用于獲取圖像實(shí)例分割任務(wù)，所述圖像實(shí)例分割任務(wù)攜帶待分割圖像以及模型標(biāo)識(shí)；

解析模塊，用于對(duì)所述模型標(biāo)識(shí)對(duì)應(yīng)的實(shí)例分割模型進(jìn)行解析，得到所述實(shí)例分割模型對(duì)應(yīng)的原模型網(wǎng)絡(luò)；

更換模塊，用于調(diào)用預(yù)設(shè)子網(wǎng)絡(luò)，根據(jù)所述預(yù)設(shè)子網(wǎng)絡(luò)對(duì)所述原模型網(wǎng)絡(luò)進(jìn)行更換，得到目標(biāo)模型網(wǎng)絡(luò)；

剪枝模塊，用于對(duì)所述目標(biāo)模型網(wǎng)絡(luò)中的所述預(yù)設(shè)子網(wǎng)絡(luò)進(jìn)行剪枝處理，得到剪枝處理后的實(shí)例分割模型；

預(yù)測(cè)模塊，用于將所述待分割圖像輸入至所述剪枝處理后的實(shí)例分割模型，對(duì)所述待分割圖像進(jìn)行預(yù)測(cè)運(yùn)算，得到圖像實(shí)例分割結(jié)果。

9.根據(jù)權(quán)利要求8所述的裝置，其特征在于，所述剪枝模塊還用于獲取與所述預(yù)設(shè)子網(wǎng)絡(luò)對(duì)應(yīng)的通道縮放因子；根據(jù)所述通道縮放因子將所述目標(biāo)模型網(wǎng)絡(luò)中所述預(yù)設(shè)子網(wǎng)絡(luò)的網(wǎng)絡(luò)通道進(jìn)行排列；對(duì)所述排列后的網(wǎng)絡(luò)通道進(jìn)行剪枝處理，得到剪枝處理后的實(shí)例分割模型。