[發(fā)明專利]一種短視頻可見(jiàn)水印去除的方法及計(jì)算機(jī)可讀存儲(chǔ)介質(zhì)

在審

• 申請(qǐng)?zhí)枺?/td>: 202210311573.0
• 申請(qǐng)日: 2022-03-28
• 公開(kāi)（公告）號(hào): CN114862642A
• 公開(kāi)（公告）日: 2022-08-05
• 發(fā)明（設(shè)計(jì)）人: 劉懷亮;張曉瑾;楊斌;趙艦波;張善莊;王亞凱
• 申請(qǐng)（專利權(quán)）人: 西安知了科技有限公司
• 主分類號(hào): G06T1/00
• 分類號(hào): G06T1/00;G06T5/00;G06T7/11;G06T7/136;G06T7/194;G06N3/04;G06N3/08
• 代理公司: 西安嘉思特知識(shí)產(chǎn)權(quán)代理事務(wù)所(普通合伙) 61230
• 代理人: 方婷

1.一種短視頻可見(jiàn)水印去除的方法，其特征在于，所述短視頻可見(jiàn)水印去除的方法包括：

步驟1、獲取待去除水印的目標(biāo)視頻的若干視頻幀；

步驟2、利用訓(xùn)練好的YOLO網(wǎng)絡(luò)結(jié)構(gòu)得到所述視頻幀水印區(qū)域的預(yù)測(cè)框；

步驟3、利用閾值分割法對(duì)所述預(yù)測(cè)框中的水印區(qū)域進(jìn)行分割，得到水印分割圖像；

步驟4、利用Criminisi算法去除所述水印分割圖像中的水印，得到去除水印后的圖像；

步驟5、將所有去除水印后的圖像視頻幀轉(zhuǎn)為視頻，得到無(wú)水印的目標(biāo)視頻。

2.根據(jù)權(quán)利要求1所述的短視頻可見(jiàn)水印去除的方法，其特征在于，所述步驟2包括：

步驟2.1、將預(yù)設(shè)大小的每張所述視頻幀切割為若干單元格，得到切割好的視頻幀圖像；

步驟2.2、將所述視頻幀圖像輸入到訓(xùn)練好的YOLO網(wǎng)絡(luò)結(jié)構(gòu)中，輸出預(yù)設(shè)數(shù)量的張量，所述張量的大小為S×S(B×5+C+E)，其中，S為行、列單元格數(shù)，B為每一所述單元格的預(yù)測(cè)框數(shù)量，5為五個(gè)特征參數(shù)，C為水印類別，E為類別存在判斷；

步驟2.3、根據(jù)所述預(yù)測(cè)框的水印類別預(yù)測(cè)概率與所述預(yù)測(cè)框的置信度的乘積得到所述預(yù)測(cè)框的得分；

步驟2.4、基于所述預(yù)測(cè)框的得分，使用NMS算法去除多余預(yù)測(cè)框，得到最終保留的預(yù)測(cè)框。

3.根據(jù)權(quán)利要求2所述的短視頻可見(jiàn)水印去除的方法，其特征在于，對(duì)于步驟2.2，若檢測(cè)的水印存在于水印數(shù)據(jù)集中，則E＝1，若不存在，則E＝0，并且將出現(xiàn)的新水印類別添加到水印數(shù)據(jù)集中，然后利用更新后的水印數(shù)據(jù)集重新訓(xùn)練YOLO網(wǎng)絡(luò)結(jié)構(gòu)，其中，所述水印數(shù)據(jù)集為用于訓(xùn)練YOLO網(wǎng)絡(luò)結(jié)構(gòu)的數(shù)據(jù)集。

4.根據(jù)權(quán)利要求2所述的短視頻可見(jiàn)水印去除的方法，其特征在于，所述步驟2.4包括：

步驟2.41、將所有預(yù)測(cè)框的得分按照從高到低的順序排序，并選取最高得分的預(yù)測(cè)框A；

步驟2.42、計(jì)算所述預(yù)測(cè)框A與其他預(yù)測(cè)框的IoU值，若IoU值大于或者等于0.5，則將該與預(yù)測(cè)框A對(duì)應(yīng)的預(yù)測(cè)框的得分設(shè)為0，所述預(yù)測(cè)框A得分不變，若IoU值小于0.5，則保持所述預(yù)測(cè)框A與其他預(yù)測(cè)框的得分不變；

步驟2.43、按照從高到低的順序，從剩余的預(yù)測(cè)框中，選取得分最高的預(yù)測(cè)框B，重復(fù)步驟2.42，直至最后保留得分不為0的預(yù)測(cè)框，作為最終保留的預(yù)測(cè)框。

5.根據(jù)權(quán)利要求1所述的短視頻可見(jiàn)水印去除的方法，其特征在于，所述YOLO網(wǎng)絡(luò)結(jié)構(gòu)的訓(xùn)練方法包括：

S1、獲取水印圖像數(shù)據(jù)集，所述水印圖像數(shù)據(jù)集包括若干水印圖像；

S2、使用labelImg工具對(duì)水印圖像進(jìn)行真實(shí)目標(biāo)框標(biāo)注，得到水印標(biāo)注圖像，水印標(biāo)注圖像包括目標(biāo)框的位置、水印類別、水印路徑，所有水印標(biāo)注圖像組成水印數(shù)據(jù)集；

S3、將預(yù)設(shè)大小的水印標(biāo)注圖像切割為若干單元格，并將切割好的水印標(biāo)注圖像輸入至所述YOLO網(wǎng)絡(luò)結(jié)構(gòu)中，輸出預(yù)設(shè)數(shù)量的張量；

S4、通過(guò)輸出的向量計(jì)算損失值，直至損失值達(dá)到最小值時(shí)，得到訓(xùn)練好的YOLO網(wǎng)絡(luò)結(jié)構(gòu)。

6.根據(jù)權(quán)利要求1所述的短視頻可見(jiàn)水印去除的方法，其特征在于，所述步驟3包括：

步驟3.1、根據(jù)所述預(yù)測(cè)框中的水印區(qū)域，得到最終閾值T_K，其中，閾值為最終閾值T_K時(shí)，所述前景的平均灰度值和所述背景的平均灰度值相等；

步驟3.2、以所述最終閾值T_K對(duì)所述預(yù)測(cè)框中的水印區(qū)域進(jìn)行分割，得到所述水印分割圖像。