一種基于機(jī)器學(xué)習(xí)的噪聲圖像顯著性檢測(cè)方法
【專利摘要】本發(fā)明涉及一種基于機(jī)器學(xué)習(xí)的噪聲圖像顯著性檢測(cè)方法,包括以下步驟:1、對(duì)每個(gè)幅度的噪聲圖像分別采用多種去噪?yún)?shù),獲得每個(gè)幅度的最佳去噪?yún)?shù);2、對(duì)每幅噪聲圖像使用噪聲評(píng)估算法進(jìn)行特征提取,獲得噪聲值特征,組成噪聲值特征集;3、將噪聲值特征集作為機(jī)器學(xué)習(xí)算法的特征集,并通過(guò)機(jī)器學(xué)習(xí)算法和五等分交叉驗(yàn)證方法,獲得噪聲幅度預(yù)測(cè)模型;4、采用噪聲幅度預(yù)測(cè)模型對(duì)相應(yīng)的噪聲圖像進(jìn)行預(yù)測(cè),獲得預(yù)測(cè)噪聲幅度值;5、采用每幅噪聲圖像的預(yù)測(cè)噪聲幅度值和對(duì)應(yīng)的最佳去噪?yún)?shù)進(jìn)行去噪處理,獲得去噪圖像集;6、對(duì)去噪圖像集中的圖像使用顯著性檢測(cè)方法,獲得最終的顯著性圖。該方法可提高顯著性檢測(cè)方法在噪聲圖像中的檢測(cè)性能。
【專利說(shuō)明】
一種基于機(jī)器學(xué)習(xí)的噪聲圖像顯著性檢測(cè)方法
技術(shù)領(lǐng)域
[0001] 本發(fā)明涉及圖像和視頻處理以及計(jì)算機(jī)視覺(jué)技術(shù)領(lǐng)域,特別是一種基于機(jī)器學(xué)習(xí) 的噪聲圖像顯著性檢測(cè)方法。
【背景技術(shù)】
[0002] 人類感官主要包括視覺(jué)、嗅覺(jué)、味覺(jué)、聽(tīng)覺(jué)和觸覺(jué)。人類依賴感官來(lái)接受外界傳遞 的信息。視覺(jué)感官在人類的感官中占了很重要的地位。人類的視覺(jué)系統(tǒng)能夠在短時(shí)間內(nèi)將 注意力關(guān)注在圖像中最為重要的部分,也就是人眼最為感興趣的部分。隨著多媒體時(shí)代的 到來(lái),各種數(shù)碼產(chǎn)品的普及和網(wǎng)絡(luò)時(shí)代數(shù)字化圖像的傳播,每天都產(chǎn)生和傳遞著大量的圖 像資源。海量的圖像數(shù)據(jù)雖然豐富了生活,但也帶來(lái)了不少挑戰(zhàn)。
[0003] 如何能夠高效且準(zhǔn)確的處理這些圖像資源是一個(gè)很關(guān)鍵的問(wèn)題。研究人員發(fā)現(xiàn)了 人類視覺(jué)系統(tǒng)的選擇性注意機(jī)制后,試圖讓計(jì)算機(jī)模擬人類視覺(jué)系統(tǒng),從而提出了顯著性 檢測(cè)方法。顯著性檢測(cè)已經(jīng)應(yīng)用到圖像壓縮與編碼、圖像檢索、圖像分割、目標(biāo)識(shí)別和內(nèi)容 感知圖像縮放等。如在圖像壓縮與編碼中,首先檢測(cè)出顯著區(qū)域,然后對(duì)顯著區(qū)域保留更多 的細(xì)節(jié),這樣既壓縮了圖像,又能保留更多重要的細(xì)節(jié)。
[0004] 視覺(jué)顯著性檢測(cè)已經(jīng)得到了比較好的研究,然而大多數(shù)顯著性檢測(cè)模型是針對(duì)無(wú) 失真圖像提出的,并且實(shí)驗(yàn)數(shù)據(jù)是無(wú)失真圖像集合。少數(shù)論文注意到了失真圖像對(duì)顯著性 檢測(cè)的影響。Zhang等人發(fā)現(xiàn)噪聲、模糊和壓縮改變了圖像低層特征,提出了基于圖像低層 特征的自底向上的顯著性檢測(cè)模型。同時(shí),Zhang等人發(fā)現(xiàn)圖像質(zhì)量失真會(huì)引起顯著性圖的 變化,并且顯著性圖的改變和主觀的圖像質(zhì)量評(píng)估之間存在一定的聯(lián)系。Gide和Karam在 圖像質(zhì)量評(píng)估的眼動(dòng)數(shù)據(jù)集上評(píng)估了 5種顯著性檢測(cè)模型,被評(píng)估的失真類型包含模糊、噪 聲和JPEG壓縮失真。Mittal等人對(duì)圖像亮度和對(duì)比度等低層特征進(jìn)行提取,并基于這些特 征采用機(jī)器學(xué)習(xí)框架預(yù)測(cè)JPEG失真圖像的顯著性區(qū)域。Kim和Milanfar針對(duì)噪聲圖像提出 了基于非參數(shù)回歸框架的顯著性檢測(cè)模型。
[0005] 實(shí)際生活中的圖像大多是帶有失真的,如由相機(jī)傳感器、圖像處理器等外設(shè)造成 的失真、拍照設(shè)備抖動(dòng)造成的抖動(dòng)失真和圖像壓縮失真等。為了提高顯著性檢測(cè)方法在噪 聲圖像上的應(yīng)用,本發(fā)明提出一種基于機(jī)器學(xué)習(xí)的噪聲圖像顯著性檢測(cè)方法。
【發(fā)明內(nèi)容】
[0006] 本發(fā)明的目的在于提供一種基于機(jī)器學(xué)習(xí)的噪聲圖像顯著性檢測(cè)方法,該方法可 以提高顯著性檢測(cè)方法在噪聲圖像中的檢測(cè)性能。
[0007] 為實(shí)現(xiàn)上述目的,本發(fā)明的技術(shù)方案是:一種基于機(jī)器學(xué)習(xí)的噪聲圖像顯著性檢 測(cè)方法,包括以下步驟: 步驟S1:對(duì)每個(gè)幅度的噪聲圖像分別采用多種去噪?yún)?shù)進(jìn)行去噪處理,獲得每個(gè)幅度 相應(yīng)的最佳去噪?yún)?shù); 步驟S2:對(duì)每幅噪聲圖像使用噪聲評(píng)估算法進(jìn)行特征提取,獲得每幅噪聲圖像的噪聲 值特征,以此組成噪聲值特征集p; 步驟S3:將噪聲值特征集P作為機(jī)器學(xué)習(xí)算法的特征集,并通過(guò)機(jī)器學(xué)習(xí)算法和五等分 交叉驗(yàn)證方法,獲得噪聲圖像的噪聲幅度預(yù)測(cè)模型; 步驟S4:采用噪聲幅度預(yù)測(cè)模型對(duì)相應(yīng)的噪聲圖像進(jìn)行預(yù)測(cè),獲得每幅噪聲圖像的預(yù) 測(cè)噪聲幅度值; 步驟S5:采用每幅噪聲圖像的預(yù)測(cè)噪聲幅度值和該幅度對(duì)應(yīng)的最佳去噪?yún)?shù)進(jìn)行去噪 處理,獲得去噪圖像集; 步驟S6:對(duì)去噪圖像集中的圖像使用顯著性檢測(cè)方法進(jìn)行檢測(cè),獲得最終的顯著性圖。
[0008] 進(jìn)一步地,所述步驟S1中,對(duì)每個(gè)幅度的噪聲圖像分別采用多種去噪?yún)?shù)進(jìn)行去 噪處理,獲得每個(gè)幅度相應(yīng)的最佳去噪?yún)?shù),具體包括以下步驟: 步驟S11:使用種高斯低通濾波去噪?yún)?shù)對(duì)每個(gè)幅度的噪聲圖像進(jìn)行去噪處理,獲得 每個(gè)幅度含種去噪?yún)?shù)的去噪后圖像集合S; 步驟S12:對(duì)去噪后圖像集合S使用顯著性檢測(cè)方法VA計(jì)算顯著性圖,獲得去噪后圖像 的顯著性圖集合T; 步驟S13:使用評(píng)價(jià)指標(biāo)PR-AUC對(duì)去噪后圖像的顯著性圖集合T進(jìn)行評(píng)估,針對(duì)每個(gè)幅 度找出平均PR-AUC最高值時(shí)使用的去噪?yún)?shù),得到每個(gè)幅度的最佳去噪?yún)?shù)。
[0009] 進(jìn)一步地,所述步驟S2中,對(duì)每幅噪聲圖像使用噪聲評(píng)估算法進(jìn)行特征提取,獲得 每幅噪聲圖像的噪聲值特征,以此組成噪聲值特征集八具體包括以下步驟: 步驟S21:對(duì)噪聲圖像進(jìn)行灰度化處理,得到灰度圖像I; 步驟S22:使用雙邊濾波處理灰度圖像I,得到雙邊濾波結(jié)果圖f; 步驟S23:計(jì)算灰度圖像I和雙邊濾波結(jié)果圖f的差值,得到差值圖像認(rèn) 步驟S24:對(duì)灰度圖像I使用Canny邊緣檢測(cè)方法得到邊緣圖像凡對(duì)邊緣圖像i?使用膨脹 算子擴(kuò)大邊緣區(qū)域,得到擴(kuò)大的邊緣圖像 步驟S25:計(jì)算噪聲大小評(píng)估值圖像#,計(jì)算公式為:
其中,隊(duì)表示灰度圖像中像素r的值,t表示像素點(diǎn),^表示擴(kuò)大的邊緣圖像,A表示擴(kuò) 大的邊緣圖像i中像素點(diǎn)t的值為0的集合; 步驟S26:將噪聲大小評(píng)估值圖像麗均勻劃分為3X3的網(wǎng)格區(qū)域,分別計(jì)算噪聲大小評(píng) 估值圖像^圖及每個(gè)網(wǎng)格區(qū)域噪聲大小評(píng)估值,計(jì)算公式為:
其中,備表示相對(duì)應(yīng)的區(qū)域,i-1,2,…,10分別表示全圖和9個(gè)網(wǎng)格區(qū)域,表示在 區(qū)域r中像素r的值;計(jì)算得到噪聲值特征集/MA,乃,…,Ao}。
[0010] 進(jìn)一步地,所述步驟S3中,將噪聲值特征集P作為機(jī)器學(xué)習(xí)算法的特征集,并通過(guò) 機(jī)器學(xué)習(xí)算法和五等分交叉驗(yàn)證方法,獲得噪聲圖像的噪聲幅度預(yù)測(cè)模型,具體包括以下 步驟: 步驟S31:將噪聲值特征集P中特征值A(chǔ),乃,…,Ao按從小到大順序排列后,作為機(jī) 器學(xué)習(xí)算法的特征集F,并將特征集F隨機(jī)五等分為:FI、F2、F3、F4和F5; 步驟S32:將F2、F3、F4和F5作為機(jī)器學(xué)習(xí)的訓(xùn)練數(shù)據(jù)集,將其對(duì)應(yīng)在圖像質(zhì)量評(píng)估數(shù)據(jù) 庫(kù)的圖像失真幅度作為機(jī)器學(xué)習(xí)的訓(xùn)練標(biāo)簽,學(xué)習(xí)得到噪聲幅度預(yù)測(cè)模型Ml; 步驟S33:重復(fù)步驟S32,分別求出F1、F3、F4和F5作為訓(xùn)練數(shù)據(jù)集時(shí)的噪聲幅度預(yù)測(cè)模 型M2,F(xiàn)1、F2、F4和F5作為訓(xùn)練數(shù)據(jù)集時(shí)的噪聲幅度預(yù)測(cè)模型M3,F(xiàn)1、F2、F3和F5作為訓(xùn)練數(shù) 據(jù)集時(shí)的噪聲幅度預(yù)測(cè)模型14^^2、?3和?4作為訓(xùn)練數(shù)據(jù)集時(shí)的噪聲幅度預(yù)測(cè)模型15。
[0011] 進(jìn)一步地,所述步驟S4中,采用噪聲幅度預(yù)測(cè)模型對(duì)相應(yīng)的噪聲圖像進(jìn)行預(yù)測(cè),獲 得每幅噪聲圖像的預(yù)測(cè)噪聲幅度值,具體包括以下步驟: 步驟S41:采用噪聲幅度預(yù)測(cè)模型Ml對(duì)特征集F1對(duì)應(yīng)的圖像集進(jìn)行預(yù)測(cè),求出噪聲幅度 值預(yù)測(cè)集合VI; 步驟S42:重復(fù)步驟S41的方法,分別采用噪聲幅度預(yù)測(cè)模型M2、M3、M4、M5對(duì)特征集F2、 F3、F4和F5對(duì)應(yīng)的圖像集預(yù)測(cè),得到噪聲幅度值預(yù)測(cè)集合V2、V3、V4、V5; 步驟S43:綜合噪聲幅度值預(yù)測(cè)集合¥={¥1、¥2、¥3、¥4、¥5},得到完整的圖像集的噪聲幅 度值預(yù)測(cè)集合V。
[0012] 進(jìn)一步地,所述步驟S5中,采用每幅噪聲圖像的預(yù)測(cè)噪聲幅度值和該幅度對(duì)應(yīng)的 最佳去噪?yún)?shù)進(jìn)行去噪處理,獲得去噪圖像集,具體包括以下步驟: 步驟S51:針對(duì)每幅噪聲圖像,從噪聲幅度值預(yù)測(cè)集合V中找到該噪聲圖像對(duì)應(yīng)的噪聲 幅度值; 步驟S52:根據(jù)噪聲幅度值,采用相對(duì)應(yīng)的最佳去噪?yún)?shù)對(duì)噪聲圖像使用高斯低通濾波 處理,獲得去噪圖像集FI。
[0013] 相較于現(xiàn)有技術(shù),本發(fā)明的有益效果是:首先利用機(jī)器學(xué)習(xí)預(yù)測(cè)噪聲圖像的噪聲 幅度,然后使用適合該幅度的最佳去噪?yún)?shù)進(jìn)行去噪處理,最后采用顯著性檢測(cè)方法計(jì)算 去噪后圖像的顯著性圖,由于本發(fā)明考慮到噪聲圖像對(duì)顯著性檢測(cè)方法的影響,因此能夠 有效的提高顯著性檢測(cè)方法在噪聲圖像上的檢測(cè)性能,可應(yīng)用于圖像和視頻處理、計(jì)算機(jī) 視覺(jué)等諸多領(lǐng)域。
【附圖說(shuō)明】
[0014] 圖1是本發(fā)明方法的流程框圖。
[0015] 圖2是本發(fā)明一實(shí)施例的步驟S2中的示例圖片(為更好的顯示效果,圖2中的(d)、 (g )和(h )的像素值被映射到[0,1 ])。
[0016] 圖3是本發(fā)明一實(shí)施例的整體方法的實(shí)現(xiàn)流程圖。
[0017] 圖4是本發(fā)明一實(shí)施例中原噪聲圖像和經(jīng)過(guò)步驟S5和S6的最終效果示例圖片。
【具體實(shí)施方式】
[0018] 下面結(jié)合附圖及具體實(shí)施例對(duì)本發(fā)明作進(jìn)一步的詳細(xì)說(shuō)明。
[0019] 本發(fā)明提供一種基于機(jī)器學(xué)習(xí)的噪聲圖像顯著性檢測(cè)方法,如圖1和圖3所示,包 括以下步驟: 步驟S1:對(duì)每個(gè)幅度的噪聲圖像分別采用多種去噪?yún)?shù)進(jìn)行去噪處理,獲得每個(gè)幅度 相應(yīng)的最佳去噪?yún)?shù)。在本實(shí)施例中,步驟S1具體包括以下步驟: 步驟S11:使用9種高斯低通濾波去噪?yún)?shù)(模板尺寸分別為{3X3, 5X5, 7X7},標(biāo)準(zhǔn) 差分別為{0.5, 0.7, 0.9})對(duì)每個(gè)幅度的噪聲圖像進(jìn)行去噪處理,獲得每個(gè)幅度含9種去 噪?yún)?shù)的去噪后圖像集合S; 步驟S12:對(duì)去噪后圖像集合S使用顯著性檢測(cè)方法VA(Saliency detection via absorbing markov chain)計(jì)算顯著性圖,獲得去噪后圖像的顯著性圖集合T; 步驟S13:使用評(píng)價(jià)指標(biāo)PR_AUC(the area under precision-recall curve)對(duì)去噪后 圖像的顯著性圖集合T進(jìn)行評(píng)估,針對(duì)每個(gè)幅度找出平均PR-AUC最高值時(shí)使用的去噪?yún)?shù), 得到每個(gè)幅度的最佳去噪?yún)?shù)。
[0020]步驟S2:對(duì)每幅噪聲圖像使用噪聲評(píng)估算法進(jìn)行特征提取,獲得每幅噪聲圖像的 噪聲值特征,以此組成噪聲值特征集八在本實(shí)施例中,如圖2所示,步驟S2具體包括以下步 驟: 步驟S21:對(duì)噪聲圖像進(jìn)行灰度化處理,得到灰度圖像1(如圖2(b)); 步驟S22:使用雙邊濾波處理灰度圖像I,得到雙邊濾波結(jié)果圖/ (如圖2(c)); 步驟S23:計(jì)算灰度圖像I和雙邊濾波結(jié)果圖j的差值,得到差值圖像漢如圖2(d)); 步驟S24:對(duì)灰度圖像I使用Canny邊緣檢測(cè)方法得到邊緣圖像K如圖2(e)),對(duì)邊緣圖 像i?使用膨脹算子擴(kuò)大邊緣區(qū)域,得到擴(kuò)大的邊緣圖像|(如圖2(f)); 步驟S25:計(jì)算噪聲大小評(píng)估值圖像#,計(jì)算公式為:
其中,隊(duì)表示灰度圖像中像素r的值,t表示像素點(diǎn),^表示擴(kuò)大的邊緣圖像,A表示擴(kuò) 大的邊緣圖像i中像素點(diǎn)t的值為0的集合; 步驟S 2 6 :將噪聲大小評(píng)估值圖像#(如圖2 (g))均勻劃分為3 X 3的網(wǎng)格區(qū)域(如圖2 (h)),分別計(jì)算噪聲大小評(píng)估值圖像^圖及每個(gè)網(wǎng)格區(qū)域噪聲大小評(píng)估值,計(jì)算公式為:
其中,備表示相對(duì)應(yīng)的區(qū)域,i-1,2,…,10分別表示全圖和9個(gè)網(wǎng)格區(qū)域,表示在 區(qū)域r中像素r的值;計(jì)算得到噪聲值特征集/MA,乃,…,Ao}。
[0021]步驟S3:將噪聲值特征集P作為機(jī)器學(xué)習(xí)算法的特征集,并通過(guò)機(jī)器學(xué)習(xí)算法和五 等分交叉驗(yàn)證方法,獲得噪聲圖像的噪聲幅度預(yù)測(cè)模型。在本實(shí)施例中,步驟S3具體包括以 下步驟: 步驟S31:將噪聲值特征集P中特征值A(chǔ),乃,…,Ao按從小到大順序排列后,作為機(jī) 器學(xué)習(xí)算法的特征集F,并將特征集F隨機(jī)五等分為:FI、F2、F3、F4和F5; 步驟S32:將F2、F3、F4和F5作為機(jī)器學(xué)習(xí)的訓(xùn)練數(shù)據(jù)集,將其對(duì)應(yīng)在圖像質(zhì)量評(píng)估數(shù)據(jù) 庫(kù)TID2013的圖像失真幅度作為機(jī)器學(xué)習(xí)的訓(xùn)練標(biāo)簽,學(xué)習(xí)得到噪聲幅度預(yù)測(cè)模型Ml; 步驟S33:重復(fù)步驟S32,分別求出F1、F3、F4和F5作為訓(xùn)練數(shù)據(jù)集時(shí)的噪聲幅度預(yù)測(cè)模 型M2,F(xiàn)1、F2、F4和F5作為訓(xùn)練數(shù)據(jù)集時(shí)的噪聲幅度預(yù)測(cè)模型M3,F(xiàn)1、F2、F3和F5作為訓(xùn)練數(shù) 據(jù)集時(shí)的噪聲幅度預(yù)測(cè)模型14^^2、?3和?4作為訓(xùn)練數(shù)據(jù)集時(shí)的噪聲幅度預(yù)測(cè)模型15。
[0022] 步驟S4:采用噪聲幅度預(yù)測(cè)模型對(duì)相應(yīng)的噪聲圖像進(jìn)行預(yù)測(cè),獲得每幅噪聲圖像 的預(yù)測(cè)噪聲幅度值。在本實(shí)施例中,步驟S4具體包括以下步驟: 步驟S41:采用噪聲幅度預(yù)測(cè)模型Ml對(duì)特征集F1對(duì)應(yīng)的圖像集進(jìn)行預(yù)測(cè),求出噪聲幅度 值預(yù)測(cè)集合VI; 步驟S42:重復(fù)步驟S41的方法,分別采用噪聲幅度預(yù)測(cè)模型M2、M3、M4、M5對(duì)特征集F2、 F3、F4和F5對(duì)應(yīng)的圖像集預(yù)測(cè),得到噪聲幅度值預(yù)測(cè)集合V2、V3、V4、V5; 步驟S43:綜合噪聲幅度值預(yù)測(cè)集合¥={¥1、¥2、¥3、¥4、¥5},得到完整的圖像集的噪聲幅 度值預(yù)測(cè)集合V。
[0023] 步驟S4:采用噪聲幅度預(yù)測(cè)模型對(duì)相應(yīng)的噪聲圖像進(jìn)行預(yù)測(cè),獲得每幅噪聲圖像 的預(yù)測(cè)噪聲幅度值。具體包括以下步驟: 步驟S41:采用噪聲幅度預(yù)測(cè)模型Ml對(duì)特征集F1對(duì)應(yīng)的圖像集進(jìn)行預(yù)測(cè),求出噪聲幅度 值預(yù)測(cè)集合VI; 步驟S42:重復(fù)步驟S41的方法,分別采用噪聲幅度預(yù)測(cè)模型M2、M3、M4、M5對(duì)特征集F2、 F3、F4和F5對(duì)應(yīng)的圖像集預(yù)測(cè),得到噪聲幅度值預(yù)測(cè)集合V2、V3、V4、V5; 步驟S43:綜合噪聲幅度值預(yù)測(cè)集合¥={¥1、¥2、¥3、¥4、¥5},得到完整的圖像集的噪聲幅 度值預(yù)測(cè)集合V。
[0024]步驟S5:采用每幅噪聲圖像的預(yù)測(cè)噪聲幅度值和該幅度對(duì)應(yīng)的最佳去噪?yún)?shù)進(jìn)行 去噪處理,獲得去噪圖像集。在本實(shí)施例中,如圖4所示,步驟S5具體包括以下步驟: 步驟S51:針對(duì)每幅噪聲圖像,從噪聲幅度值預(yù)測(cè)集合V中找到該噪聲圖像對(duì)應(yīng)的噪聲 幅度值; 步驟S52:根據(jù)噪聲幅度值,采用相對(duì)應(yīng)的最佳去噪?yún)?shù)對(duì)噪聲圖像使用高斯低通濾波 處理,獲得去噪圖像集FI。
[0025]步驟S6:對(duì)去噪圖像集FI中的圖像使用顯著性檢測(cè)方法VA進(jìn)行檢測(cè),獲得最終的 顯著性圖。
[0026]本發(fā)明提供的基于機(jī)器學(xué)習(xí)的噪聲圖像顯著性檢測(cè)方法,考慮到噪聲圖像對(duì)顯著 性檢測(cè)方法的影響,挖掘噪聲大小評(píng)估值特征與圖像質(zhì)量評(píng)估數(shù)據(jù)庫(kù)TID2013中的噪聲幅 度的關(guān)聯(lián),設(shè)計(jì)得到機(jī)器學(xué)習(xí)噪聲預(yù)測(cè)模型,并結(jié)合去噪方法和為該幅度設(shè)置的參數(shù)對(duì)圖 像進(jìn)行去噪處理,最后采用顯著性檢測(cè)方法VA計(jì)算去噪后圖像的顯著性圖。該方法能夠有 效的提高顯著性檢測(cè)方法在噪聲圖像上的檢測(cè)性能,可應(yīng)用于圖像和視頻處理、計(jì)算機(jī)視 覺(jué)等領(lǐng)域。
[0027]以上是本發(fā)明的較佳實(shí)施例,凡依本發(fā)明技術(shù)方案所作的改變,所產(chǎn)生的功能作 用未超出本發(fā)明技術(shù)方案的范圍時(shí),均屬于本發(fā)明的保護(hù)范圍。
【主權(quán)項(xiàng)】
1. 一種基于機(jī)器學(xué)習(xí)的噪聲圖像顯著性檢測(cè)方法,其特征在于,包括W下步驟: 步驟SI:對(duì)每個(gè)幅度的噪聲圖像分別采用多種去噪?yún)?shù)進(jìn)行去噪處理,獲得每個(gè)幅度 相應(yīng)的最佳去噪?yún)?shù); 步驟S2:對(duì)每幅噪聲圖像使用噪聲評(píng)估算法進(jìn)行特征提取,獲得每幅噪聲圖像的噪聲 值特征,W此組成噪聲值特征集戶; 步驟S3:將噪聲值特征集戶作為機(jī)器學(xué)習(xí)算法的特征集,并通過(guò)機(jī)器學(xué)習(xí)算法和五等分 交叉驗(yàn)證方法,獲得噪聲圖像的噪聲幅度預(yù)測(cè)模型; 步驟S4:采用噪聲幅度預(yù)測(cè)模型對(duì)相應(yīng)的噪聲圖像進(jìn)行預(yù)測(cè),獲得每幅噪聲圖像的預(yù) 測(cè)噪聲幅度值; 步驟S5:采用每幅噪聲圖像的預(yù)測(cè)噪聲幅度值和該幅度對(duì)應(yīng)的最佳去噪?yún)?shù)進(jìn)行去噪 處理,獲得去噪圖像集; 步驟S6:對(duì)去噪圖像集中的圖像使用顯著性檢測(cè)方法進(jìn)行檢測(cè),獲得最終的顯著性圖。2. 根據(jù)權(quán)利要求1所述的一種基于機(jī)器學(xué)習(xí)的噪聲圖像顯著性檢測(cè)方法,其特征在于: 所述步驟Sl中,對(duì)每個(gè)幅度的噪聲圖像分別采用多種去噪?yún)?shù)進(jìn)行去噪處理,獲得每個(gè)幅 度相應(yīng)的最佳去噪?yún)?shù),具體包括W下步驟: 步驟Sll:使用n種高斯低通濾波去噪?yún)?shù)對(duì)每個(gè)幅度的噪聲圖像進(jìn)行去噪處理,獲得 每個(gè)幅度含n種去噪?yún)?shù)的去噪后圖像集合S; 步驟S12:對(duì)去噪后圖像集合S使用顯著性檢測(cè)方法VA計(jì)算顯著性圖,獲得去噪后圖像 的顯著性圖集合T; 步驟S13:使用評(píng)價(jià)指標(biāo)PR-AUC對(duì)去噪后圖像的顯著性圖集合T進(jìn)行評(píng)估,針對(duì)每個(gè)幅 度找出平均PR-AUC最高值時(shí)使用的去噪?yún)?shù),得到每個(gè)幅度的最佳去噪?yún)?shù)。3. 根據(jù)權(quán)利要求1所述的一種基于機(jī)器學(xué)習(xí)的噪聲圖像顯著性檢測(cè)方法,其特征在于: 所述步驟S2中,對(duì)每幅噪聲圖像使用噪聲評(píng)估算法進(jìn)行特征提取,獲得每幅噪聲圖像的噪 聲值特征,W此組成噪聲值特征集A具體包括W下步驟: 步驟S21:對(duì)噪聲圖像進(jìn)行灰度化處理,得到灰度圖像/; 步驟S22:使用雙邊濾波處理灰度圖像/,得到雙邊濾波結(jié)果圖/; 步驟S23:計(jì)算灰度圖像巧日雙邊濾波結(jié)果圖f的差值,得到差值圖像化 步驟S24:對(duì)灰度圖像/使用化nny邊緣檢測(cè)方法得到邊緣圖像i?,對(duì)邊緣圖像i?使用膨脹 算子擴(kuò)大邊緣區(qū)域,得到擴(kuò)大的邊緣圖像£^ 步驟S25:計(jì)算噪聲大小評(píng)估值圖像#,計(jì)算公式為:其中,化表示灰度圖像中像素誠(chéng)]值,t表示像素點(diǎn),if表示擴(kuò)大的邊緣圖像,A表示擴(kuò)大 的邊緣圖像^中像素點(diǎn)t的值為O的集合; 步驟S26:將噪聲大小評(píng)估值圖像M勻勻劃分為3 X 3的網(wǎng)格區(qū)域,分別計(jì)算噪聲大小評(píng) 估值圖像始^圖及每個(gè)網(wǎng)格區(qū)域噪聲大小評(píng)估值,計(jì)算公式為: 其中,做表示相對(duì)應(yīng)的區(qū)域,1-1,Z,…,^巧刑巧不全圖和9個(gè)網(wǎng)格區(qū)域,做,*^表示在 區(qū)域r中像素沖勺值;計(jì)算得到噪聲值特征集片{A,巧,…,AoK4. 根據(jù)權(quán)利要求3所述的一種基于機(jī)器學(xué)習(xí)的噪聲圖像顯著性檢測(cè)方法,其特征在于: 所述步驟S3中,將噪聲值特征集戶作為機(jī)器學(xué)習(xí)算法的特征集,并通過(guò)機(jī)器學(xué)習(xí)算法和五等 分交叉驗(yàn)證方法,獲得噪聲圖像的噪聲幅度預(yù)測(cè)模型,具體包括W下步驟: 步驟S31:將噪聲值特征集戶+^特征值月,巧,…,月0按從小到大順序排列后,作為機(jī)器 學(xué)習(xí)算法的特征集F,并將特征集F隨機(jī)五等分為:Fl、F2、F3、F4和巧; 步驟S32:將F2、F3、F4和F5作為機(jī)器學(xué)習(xí)的訓(xùn)練數(shù)據(jù)集,將其對(duì)應(yīng)在圖像質(zhì)量評(píng)估數(shù)據(jù) 庫(kù)的圖像失真幅度作為機(jī)器學(xué)習(xí)的訓(xùn)練標(biāo)簽,學(xué)習(xí)得到噪聲幅度預(yù)測(cè)模型Ml; 步驟S33:重復(fù)步驟S32,分別求出F1、F3、F4和巧作為訓(xùn)練數(shù)據(jù)集時(shí)的噪聲幅度預(yù)測(cè)模 型M2,F(xiàn)1、F2、F4和F5作為訓(xùn)練數(shù)據(jù)集時(shí)的噪聲幅度預(yù)測(cè)模型M3,F(xiàn)1、F2、F3和F5作為訓(xùn)練數(shù) 據(jù)集時(shí)的噪聲幅度預(yù)測(cè)模型14^1少2、。3和。4作為訓(xùn)練數(shù)據(jù)集時(shí)的噪聲幅度預(yù)測(cè)模型15。5. 根據(jù)權(quán)利要求4所述的一種基于機(jī)器學(xué)習(xí)的噪聲圖像顯著性檢測(cè)方法,其特征在于: 所述步驟S4中,采用噪聲幅度預(yù)測(cè)模型對(duì)相應(yīng)的噪聲圖像進(jìn)行預(yù)測(cè),獲得每幅噪聲圖像的 預(yù)測(cè)噪聲幅度值,具體包括W下步驟: 步驟S41:采用噪聲幅度預(yù)測(cè)模型Ml對(duì)特征集Fl對(duì)應(yīng)的圖像集進(jìn)行預(yù)測(cè),求出噪聲幅度 值預(yù)測(cè)集合VI; 步驟S42:重復(fù)步驟S41的方法,分別采用噪聲幅度預(yù)測(cè)模型12、13、14、15對(duì)特征集尸2、 F3、F4和巧對(duì)應(yīng)的圖像集預(yù)測(cè),得到噪聲幅度值預(yù)測(cè)集合V2、V3、V4、V5; 步驟S43:綜合噪聲幅度值預(yù)測(cè)集合V= {Vl、V2、V3、V4、V引,得到完整的圖像集的噪聲幅 度值預(yù)測(cè)集合V。6. 根據(jù)權(quán)利要求5所述的一種基于機(jī)器學(xué)習(xí)的噪聲圖像顯著性檢測(cè)方法,其特征在于, 所述步驟S5中,采用每幅噪聲圖像的預(yù)測(cè)噪聲幅度值和該幅度對(duì)應(yīng)的最佳去噪?yún)?shù)進(jìn)行去 噪處理,獲得去噪圖像集,具體包括W下步驟: 步驟S51:針對(duì)每幅噪聲圖像,從噪聲幅度值預(yù)測(cè)集合V中找到該噪聲圖像對(duì)應(yīng)的噪聲 幅度值; 步驟S52:根據(jù)噪聲幅度值,采用相對(duì)應(yīng)的最佳去噪?yún)?shù)對(duì)噪聲圖像使用高斯低通濾波 處理,獲得去噪圖像集FI。
【文檔編號(hào)】G06T5/00GK105913427SQ201610222900
【公開(kāi)日】2016年8月31日
【申請(qǐng)日】2016年4月12日
【發(fā)明人】牛玉貞, 林樂(lè)凝, 陳羽中
【申請(qǐng)人】福州大學(xué)