[發(fā)明專利]圖像補全方法、裝置、計算機設備及存儲介質(zhì)

有效

• 申請?zhí)枺?/td>: 201810417418.0
• 申請日: 2018-05-04
• 公開（公告）號: CN108765315B
• 公開（公告）日: 2021-09-07
• 發(fā)明（設計）人: 張弓
• 申請（專利權(quán)）人: OPPO廣東移動通信有限公司
• 主分類號: G06T5/00
• 分類號: G06T5/00;G06K9/46;G06N3/04
• 代理公司: 北京恒博知識產(chǎn)權(quán)代理有限公司 11528
• 代理人: 范勝祥

1.一種圖像補全方法，其特征在于，包括以下步驟：

對待補全的第一圖像進行圖像特征提取，并根據(jù)提取的所述圖像特征，識別出所述第一圖像中的缺失區(qū)域；

對所述第一圖像中所述缺失區(qū)域周邊區(qū)域的內(nèi)容進行識別，得到所述缺失區(qū)域所缺失的內(nèi)容對應的語義；

根據(jù)所述所缺失的內(nèi)容對應的語義，對所述第一圖像中的所述缺失區(qū)域進行補全，得到補全后的第二圖像；

其中，所述根據(jù)提取的所述圖像特征，識別出所述第一圖像中的缺失區(qū)域，包括：

所述第一圖像的圖像特征包括第一圖像中每個像素點在所述第一圖像中的位置信息和第一圖像中每個像素點對應的圖像特征；

根據(jù)每個像素點的位置信息和圖像特征，識別出所述第一圖像中的缺失區(qū)域。

2.根據(jù)權(quán)利要求1所述的方法，其特征在于，所述對待補全的第一圖像進行圖像特征提取，并根據(jù)提取的所述圖像特征，識別出所述第一圖像中的缺失區(qū)域，包括：

對所述第一圖像中每個像素點進行圖像特征提取；

獲取每個像素點在所述第一圖像中的位置信息；

根據(jù)每個像素點的位置信息和圖像特征，從所有的像素點中識別出屬于所述缺失區(qū)域的像素點；

根據(jù)所述缺失區(qū)域的像素點的位置信息，確定出所述缺失區(qū)域。

3.根據(jù)權(quán)利要求2所述的方法，其特征在于，所述根據(jù)每個像素點的位置信息和圖像特征，從所有的像素點中識別出屬于所述缺失區(qū)域的像素點，包括：

根據(jù)每個像素點的位置信息，確定與所述像素點相鄰的相鄰像素點；

將所述像素點與所述相鄰的像素點的圖像特征進行比較；

從所有的像素點，識別出與相鄰像素點的圖像特征之間的差異小于預設差異閾值，且所述圖像特征滿足預設的圖像特征條件的至少一個第一像素點；

將由所述第一像素點構(gòu)成的區(qū)域內(nèi)的像素點識別為屬于所述缺失區(qū)域的像素點。

4.根據(jù)權(quán)利要求1所述的方法，其特征在于，所述對待補全的第一圖像進行圖像特征提取之前，還包括：

采集第一樣本圖像，并根據(jù)所述第一樣本圖像生成存在缺失區(qū)域的第二樣本圖像；

利用所述第一樣本圖像、所述第二樣本圖像和所述第一樣本圖像與所述第二樣本圖像之間的對應關(guān)系，構(gòu)成訓練樣本；

將所述訓練樣本輸入到構(gòu)建的神經(jīng)網(wǎng)絡中進行訓練，得到訓練好的目標神經(jīng)網(wǎng)絡。

5.根據(jù)權(quán)利要求4所述的方法，其特征在于，所述神經(jīng)網(wǎng)絡包括特征提取層、缺失區(qū)域識別層、缺失內(nèi)容識別層和缺失內(nèi)容補全層；

則所述將所述訓練樣本輸入到構(gòu)建的神經(jīng)網(wǎng)絡中進行訓練，得到訓練好的目標神經(jīng)網(wǎng)絡，包括：

將所述第一樣本圖像和第二樣本圖像輸入到所述神經(jīng)網(wǎng)絡中，利用所述神經(jīng)網(wǎng)絡中的特征提取層，提取所述第二樣本圖像的每個像素點的特征信息；所述特征信息包括圖像特征和位置信息；

將所述特征提取層提取的特征信息輸入到所述神經(jīng)網(wǎng)絡中的缺失區(qū)域識別層，得到所述第二樣本圖像的缺失區(qū)域；

將所述第二樣本圖像的缺失區(qū)域和所述特征提取層提取的特征信息，輸入到所述神經(jīng)網(wǎng)絡中的缺失內(nèi)容識別層，得到所述第二樣本圖像的缺失區(qū)域所缺失內(nèi)容對應的語義；

將所述第二樣本圖像的所述缺失內(nèi)容對應的語義，輸入到缺失內(nèi)容補全層，結(jié)合所述特征提取層提取的特征信息，對所述第二樣本圖像中的所述缺失區(qū)域進行補全，得到補全后的所述第二樣本圖像；

將補全后的所述第二樣本圖像與第一樣本圖像進行比較，獲取代價函數(shù)，如果所述代價函數(shù)最小則完成對所述神經(jīng)網(wǎng)絡的訓練，得到所述目標神經(jīng)網(wǎng)絡，如果所述代價函數(shù)未達到最小則返回，則調(diào)整所述神經(jīng)網(wǎng)絡返回繼續(xù)訓練，直到所述代價函數(shù)最小。