本技術(shù)涉及圖像處理，更具體地，涉及一種水印嵌入方法、裝置、電子設(shè)備及存儲(chǔ)介質(zhì)。

背景技術(shù)：

1、由于互聯(lián)網(wǎng)技術(shù)的極速發(fā)展，越來越多圖像以及視頻面臨被盜版和泄露的風(fēng)險(xiǎn)，這對于版權(quán)方會(huì)造成重大的損失，因此，需要加強(qiáng)對圖像以及視頻的版權(quán)保護(hù)以及泄露后的溯源，其中，水印是圖像以及視頻版權(quán)保護(hù)中重要的一環(huán)。

2、目前，可以是以是將圖像從空域轉(zhuǎn)換到頻域，在頻域上進(jìn)行信息植入來完成圖像的水印加密。但是，采用上述水方法來嵌入的水印，嵌入的水印信息對原圖像的畫質(zhì)影響較大，導(dǎo)致嵌入水印信息之后的圖像的畫質(zhì)較差。

技術(shù)實(shí)現(xiàn)思路

1、有鑒于此，本技術(shù)實(shí)施例提出了一種水印嵌入方法、裝置、電子設(shè)備及存儲(chǔ)介質(zhì)。

2、第一方面，本技術(shù)實(shí)施例提供了一種水印嵌入方法，方法包括：獲取待處理圖像中目標(biāo)圖像塊在單顏色通道下的頻域系數(shù)序列；從目標(biāo)圖像塊在單顏色通道下的頻域系數(shù)序列中選取位于預(yù)設(shè)位置范圍的目標(biāo)頻域系數(shù)序列；根據(jù)目標(biāo)頻域系數(shù)序列中不同頻域系數(shù)之間的大小關(guān)系以及水印信息中目標(biāo)圖像塊對應(yīng)的子水印信息，對目標(biāo)頻域系數(shù)序列中的頻域系數(shù)進(jìn)行調(diào)整，得到目標(biāo)頻域系數(shù)序列對應(yīng)的調(diào)整頻域系數(shù)序列；根據(jù)調(diào)整頻域系數(shù)序列，確定目標(biāo)圖像塊在單顏色通道下對應(yīng)的參考圖像塊；將待處理圖像中的目標(biāo)圖像塊和目標(biāo)圖像塊在單顏色通道下對應(yīng)的參考圖像塊進(jìn)行融合，得到待處理圖像對應(yīng)的水印圖像。

3、第二方面，本技術(shù)實(shí)施例提供了一種水印識別方法，方法包括：獲取待識別水印圖像中包括待識別水印的預(yù)測圖像塊在單顏色通道下的預(yù)測頻域系數(shù)序列；從預(yù)測頻域系數(shù)序列中選取位于預(yù)設(shè)位置范圍的目標(biāo)預(yù)測頻域系數(shù)序列；根據(jù)目標(biāo)預(yù)測頻域系數(shù)序列中各個(gè)頻域系數(shù)之間的大小關(guān)系，確定預(yù)測圖像塊在單顏色通道下的預(yù)測水印信息；根據(jù)預(yù)測圖像塊在多個(gè)單顏色通道下的預(yù)測水印信息，確定待識別水印的水印識別結(jié)果。

4、第三方面，本技術(shù)實(shí)施例提供了一種水印嵌入裝置，裝置包括：第一獲取模塊，用于獲取待處理圖像中目標(biāo)圖像塊在單顏色通道下的頻域系數(shù)序列；第一選取模塊，用于從目標(biāo)圖像塊在單顏色通道下的頻域系數(shù)序列中選取位于預(yù)設(shè)位置范圍的目標(biāo)頻域系數(shù)序列；調(diào)整模塊，用于根據(jù)目標(biāo)頻域系數(shù)序列中不同頻域系數(shù)之間的大小關(guān)系以及水印信息中目標(biāo)圖像塊對應(yīng)的子水印信息，對目標(biāo)頻域系數(shù)序列中的頻域系數(shù)進(jìn)行調(diào)整，得到目標(biāo)頻域系數(shù)序列對應(yīng)的調(diào)整頻域系數(shù)序列；第一確定模塊，用于根據(jù)調(diào)整頻域系數(shù)序列，確定目標(biāo)圖像塊在單顏色通道下對應(yīng)的參考圖像塊；融合模塊，用于將待處理圖像中的目標(biāo)圖像塊和目標(biāo)圖像塊在單顏色通道下對應(yīng)的參考圖像塊進(jìn)行融合，得到待處理圖像對應(yīng)的水印圖像。

5、可選地，調(diào)整頻域系數(shù)序列包括目標(biāo)頻域系數(shù)序列中各頻域系數(shù)對應(yīng)的調(diào)整頻域系數(shù)；調(diào)整模塊，還用于按照預(yù)設(shè)的步長遍歷目標(biāo)頻域系數(shù)序列中的頻域系數(shù)；針對相鄰兩次遍歷中在先被遍歷的第一頻域系數(shù)和在后被遍歷的第二頻域系數(shù)，若第一頻域系數(shù)與第二頻域系數(shù)之間的大小關(guān)系和子水印信息滿足調(diào)整條件，根據(jù)第一頻域系數(shù)和第二頻域系數(shù)，確定調(diào)整系數(shù)；根據(jù)調(diào)整系數(shù)分別對第一頻域系數(shù)和第二頻域系數(shù)進(jìn)行調(diào)整，獲得第一頻域系數(shù)對應(yīng)的調(diào)整頻域系數(shù)和第二頻域系數(shù)對應(yīng)的調(diào)整頻域系數(shù)。

6、可選地，調(diào)整模塊，還用于根據(jù)第一頻域系數(shù)和第二頻域系數(shù)之間的頻域系數(shù)差值，確定調(diào)整系數(shù)。

7、可選地，調(diào)整模塊，還用于根據(jù)第一頻域系數(shù)和第二頻域系數(shù)之和以及調(diào)整系數(shù)，確定調(diào)整子系數(shù)；根據(jù)第一頻域系數(shù)以及調(diào)整子系數(shù)，對第一頻域系數(shù)進(jìn)行調(diào)整，得到第一頻域系數(shù)對應(yīng)的調(diào)整頻域系數(shù)；根據(jù)第二頻域系數(shù)以及調(diào)整子系數(shù)，對第二頻域系數(shù)進(jìn)行調(diào)整，得到第二頻域系數(shù)對應(yīng)的調(diào)整頻域系數(shù)。

8、可選地，調(diào)整模塊，還用于針對相鄰兩次遍歷中在先被遍歷的第一頻域系數(shù)和在后被遍歷的第二頻域系數(shù)，若第一頻域系數(shù)與第二頻域系數(shù)之間的大小關(guān)系和子水印信息不滿足調(diào)整條件，將第一頻域系數(shù)作為第一頻域系數(shù)對應(yīng)的調(diào)整頻域系數(shù)，并將第二頻域系數(shù)作為第二頻域系數(shù)對應(yīng)的調(diào)整頻域系數(shù)。

9、可選地，融合模塊，還用于對目標(biāo)圖像塊在多個(gè)單顏色通道下對應(yīng)的參考圖像塊進(jìn)行融合，得到目標(biāo)圖像塊對應(yīng)的融合參考圖像塊；根據(jù)目標(biāo)圖像塊與目標(biāo)圖像塊對應(yīng)的融合參考圖像塊之間的差值，確定目標(biāo)圖像塊對應(yīng)的模板圖像塊；基于預(yù)置的強(qiáng)度參數(shù)，對目標(biāo)圖像塊以及目標(biāo)圖像塊對應(yīng)的模板圖像塊進(jìn)行加權(quán)求和，得到待處理圖像對應(yīng)的水印圖像。

10、可選地，融合模塊，還用于確定目標(biāo)圖像塊與目標(biāo)圖像塊對應(yīng)的融合參考圖像塊的差值，作為目標(biāo)圖像塊對應(yīng)的中間圖像塊；對目標(biāo)圖像塊對應(yīng)的中間圖像塊進(jìn)行歸一化處理，得到目標(biāo)圖像塊對應(yīng)的模板圖像塊。

11、可選地，第一確定模塊，還用于利用調(diào)整頻域系數(shù)序列替換頻域系數(shù)序列中的目標(biāo)頻域系數(shù)序列，得到替換頻域系數(shù)序列；根據(jù)替換頻域系數(shù)序列，確定調(diào)整頻域圖；將調(diào)整頻域圖向空間域進(jìn)行變換，獲得目標(biāo)圖像塊在單顏色通道下對應(yīng)的參考圖像塊。

12、第四方面，本技術(shù)實(shí)施例提供了一種水印識別裝置，裝置包括：第二獲取模塊，用于獲取待識別水印圖像中包括待識別水印的預(yù)測圖像塊在單顏色通道下的預(yù)測頻域系數(shù)序列；第二選取模塊，用于從預(yù)測頻域系數(shù)序列中選取位于預(yù)設(shè)位置范圍的目標(biāo)預(yù)測頻域系數(shù)序列；第二確定模塊，用于根據(jù)目標(biāo)預(yù)測頻域系數(shù)序列中各個(gè)頻域系數(shù)之間的大小關(guān)系，確定預(yù)測圖像塊在單顏色通道下的預(yù)測水印信息；結(jié)果確定模塊，用于根據(jù)預(yù)測圖像塊在多個(gè)單顏色通道下的預(yù)測水印信息，確定待識別水印的水印識別結(jié)果。

13、可選地，第二確定模塊，還用于按照預(yù)設(shè)的步長遍歷目標(biāo)預(yù)測頻域系數(shù)序列中預(yù)測頻域系數(shù)；針對相鄰兩次遍歷中在先被遍歷的第一預(yù)測頻域系數(shù)和在后被遍歷的第二預(yù)測頻域系數(shù)，根據(jù)第一預(yù)測頻域系數(shù)與第二預(yù)測頻域系數(shù)之間的大小關(guān)系，確定相鄰兩次遍歷對應(yīng)的初始預(yù)測水印信息；根據(jù)多個(gè)遍歷確定的多個(gè)初始預(yù)測水印信息，確定每個(gè)初始預(yù)測水印信息的占比；根據(jù)每個(gè)初始預(yù)測水印信息的占比，確定預(yù)測圖像塊在單顏色通道下的預(yù)測水印信息。

14、可選地，第二獲取模塊，還用于將待識別水印圖像輸入目標(biāo)檢測模型，得到目標(biāo)檢測模型輸出的預(yù)測位置信息以及預(yù)測位置信息的預(yù)測概率；在預(yù)測位置信息中篩選預(yù)測概率達(dá)到概率閾值的目標(biāo)預(yù)測位置信息；獲取待識別圖像中與目標(biāo)預(yù)測位置信息對應(yīng)的區(qū)域，作為預(yù)測圖像塊；確定預(yù)測圖像塊在單顏色通道下的預(yù)測頻域系數(shù)序列。

15、可選地，裝置還包括訓(xùn)練模塊，用于獲取包括樣本水印信息的樣本圖像、樣本水印信息對應(yīng)的樣本位置信息以及樣本位置信息對應(yīng)的樣本標(biāo)簽；將樣本圖像輸入初始檢測模型，得到初始檢測模型輸出的樣本預(yù)測位置信息以及樣本預(yù)測位置的樣本預(yù)測概率；根據(jù)樣本預(yù)測位置信息以及樣本位置信息，確定第一損失值；根據(jù)樣本預(yù)測概率以及樣本標(biāo)簽，確定第二損失值；根據(jù)第一損失值以及第二損失值，對初始模型進(jìn)行訓(xùn)練，得到目標(biāo)檢測模型。

16、第五方面，本技術(shù)實(shí)施例提供了一種電子設(shè)備，包括處理器以及存儲(chǔ)器；一個(gè)或多個(gè)計(jì)算機(jī)程序被存儲(chǔ)在存儲(chǔ)器中并被配置為由處理器執(zhí)行以實(shí)現(xiàn)上述的方法。

17、第六方面，本技術(shù)實(shí)施例提供了一種計(jì)算機(jī)可讀存儲(chǔ)介質(zhì)，計(jì)算機(jī)可讀存儲(chǔ)介質(zhì)中存儲(chǔ)有計(jì)算機(jī)程序，其中，在計(jì)算機(jī)程序被處理器運(yùn)行時(shí)執(zhí)行上述的方法。

18、第七方面，本技術(shù)的實(shí)施例包括一種計(jì)算機(jī)程序產(chǎn)品，其包括承載在計(jì)算機(jī)可讀介質(zhì)上的計(jì)算機(jī)程序，該計(jì)算機(jī)程序包含用于執(zhí)行上述的方法的計(jì)算機(jī)程序。

19、本技術(shù)實(shí)施例提供的一種水印嵌入方法、裝置、電子設(shè)備及存儲(chǔ)介質(zhì)，在本技術(shù)中，針對待處理圖像中的目標(biāo)圖像塊，從目標(biāo)圖像塊在單顏色通道下的頻域系數(shù)序列中選取位于預(yù)設(shè)位置范圍的目標(biāo)頻域系數(shù)序列，然后基于目標(biāo)頻域系數(shù)序列中不同頻域系數(shù)之間的大小關(guān)系以及水印信息中目標(biāo)圖像塊對應(yīng)的子水印信息，對目標(biāo)頻域系數(shù)序列中的頻域系數(shù)進(jìn)行調(diào)整，得到目標(biāo)頻域系數(shù)序列對應(yīng)的調(diào)整頻域系數(shù)序列，這樣，保證調(diào)整頻域系數(shù)序列攜帶目標(biāo)圖像塊對應(yīng)的子水印信息，進(jìn)而保證基于調(diào)整頻域系數(shù)序列所得到的參考圖像塊攜帶對應(yīng)的子水印信息，由此實(shí)現(xiàn)將子水印信息添加到目標(biāo)圖像塊中，進(jìn)而得到待處理圖像對應(yīng)的水印圖像。而且，本技術(shù)中，對待處理圖像中預(yù)設(shè)位置范圍的目標(biāo)頻域系數(shù)序列進(jìn)行調(diào)整來獲得水印圖像，而不是對待處理圖像對應(yīng)的全部的頻域系數(shù)進(jìn)行調(diào)整，對待處理圖像的頻域系數(shù)的調(diào)整量較低且調(diào)整幅度較低，降低了水印圖像中嵌入的水印信息對原待處理圖像的影響，從而提高了得到的水印圖像的畫質(zhì)。