[發(fā)明專利]圖像增強(qiáng)方法、智能設(shè)備、計(jì)算機(jī)存儲(chǔ)介質(zhì)

有效

• 申請(qǐng)?zhí)枺?/td>: 202110755171.5
• 申請(qǐng)日: 2021-07-05
• 公開（公告）號(hào): CN113436107B
• 公開（公告）日: 2023-06-20
• 發(fā)明（設(shè)計(jì)）人: 劉日升;石東祥;仲維;樊鑫;羅鐘鉉
• 申請(qǐng)（專利權(quán)）人: 鵬城實(shí)驗(yàn)室;大連理工大學(xué)
• 主分類號(hào): G06T5/00
• 分類號(hào): G06T5/00;G06T7/90;G06V10/56;G06V10/82;G06N3/0464;G06N3/08
• 代理公司: 深圳市世紀(jì)恒程知識(shí)產(chǎn)權(quán)代理事務(wù)所 44287
• 代理人: 陳小娟

本發(fā)明公開了一種圖像增強(qiáng)方法、智能設(shè)備和計(jì)算機(jī)存儲(chǔ)介質(zhì)，所述方法包括以下步驟：獲取水下初始圖像，并對(duì)所述水下初始圖像進(jìn)行顏色空間轉(zhuǎn)化；通過對(duì)比度增強(qiáng)網(wǎng)絡(luò)對(duì)轉(zhuǎn)化后的所述水下初始圖像的L通道數(shù)據(jù)進(jìn)行處理，得到第一圖像處理數(shù)據(jù)；通過顏色矯正網(wǎng)絡(luò)對(duì)轉(zhuǎn)化后的所述水下初始圖像的AB通道數(shù)據(jù)進(jìn)行處理，得到第二圖像處理數(shù)據(jù)；根據(jù)所述第一圖像處理數(shù)據(jù)和第二圖像處理數(shù)據(jù)生成增強(qiáng)圖像，解決現(xiàn)有技術(shù)中水下圖像的對(duì)比度與顏色之間存在耦合導(dǎo)致水下圖像增強(qiáng)效果差的問題，提高水下圖像增強(qiáng)效果。

技術(shù)領(lǐng)域

本發(fā)明涉及圖像處理技術(shù)領(lǐng)域，尤其涉及一種圖像增強(qiáng)方法、智能設(shè)備、計(jì)算機(jī)存儲(chǔ)介質(zhì)。

背景技術(shù)

目前，隨著計(jì)算機(jī)視覺領(lǐng)域以及水下機(jī)器人技術(shù)的不斷發(fā)展和進(jìn)步，視覺輔助在水下探測(cè)中起到越來(lái)越大的作用，提高水下獲取的圖像質(zhì)量可以更好的輔助完成水下任務(wù)，現(xiàn)有技術(shù)中，通常采用同一個(gè)網(wǎng)絡(luò)來(lái)擬合復(fù)雜多變的水下問題，水下圖像的對(duì)比度與顏色之間存在耦合關(guān)系，導(dǎo)致水下圖像增強(qiáng)效果差。

發(fā)明內(nèi)容

本發(fā)明主要目的在于提供一種圖像增強(qiáng)方法、智能設(shè)備和計(jì)算機(jī)存儲(chǔ)介質(zhì)，旨在解決水下圖像的對(duì)比度與顏色之間存在耦合導(dǎo)致水下圖像增強(qiáng)效果差的問題。

為實(shí)現(xiàn)上述目的，本發(fā)明提供一種圖像增強(qiáng)方法；在一實(shí)施例中，所述圖像增強(qiáng)方法包括以下步驟：

獲取水下初始圖像，并對(duì)所述水下初始圖像進(jìn)行顏色空間轉(zhuǎn)化；

通過對(duì)比度增強(qiáng)網(wǎng)絡(luò)對(duì)轉(zhuǎn)化后的所述水下初始圖像的L通道數(shù)據(jù)進(jìn)行處理，得到第一圖像處理數(shù)據(jù)；

通過顏色矯正網(wǎng)絡(luò)對(duì)轉(zhuǎn)化后的所述水下初始圖像的AB通道數(shù)據(jù)進(jìn)行處理，得到第二圖像處理數(shù)據(jù)；

根據(jù)所述第一圖像處理數(shù)據(jù)和第二圖像處理數(shù)據(jù)生成增強(qiáng)圖像。

在一實(shí)施例中，所述獲取水下初始圖像，并對(duì)所述水下初始圖像進(jìn)行顏色空間轉(zhuǎn)化的步驟包括：

獲取水下初始圖像；

將所述水下初始圖像從RGB顏色空間轉(zhuǎn)化為L(zhǎng)AB顏色空間。

在一實(shí)施例中，所述將所述水下初始圖像從RGB顏色空間轉(zhuǎn)化為L(zhǎng)AB顏色空間的步驟包括：

將所述水下初始圖像從RGB顏色空間轉(zhuǎn)化為XYZ顏色空間；

將所述XYZ顏色空間轉(zhuǎn)化為L(zhǎng)AB顏色空間。

在一實(shí)施例中，所述通過對(duì)比度增強(qiáng)網(wǎng)絡(luò)對(duì)轉(zhuǎn)化后的所述水下初始圖像的L通道數(shù)據(jù)進(jìn)行處理，得到第一圖像處理數(shù)據(jù)的步驟包括：

將所述水下初始圖像的L通道數(shù)據(jù)輸入對(duì)比度增強(qiáng)網(wǎng)絡(luò)，其中，所述對(duì)比度增強(qiáng)網(wǎng)絡(luò)根據(jù)所述水下初始圖像的L通道數(shù)據(jù)確定圖像特征；

將所述圖像特征與所述對(duì)比度增強(qiáng)網(wǎng)絡(luò)中的全連接層進(jìn)行卷積，得到增強(qiáng)后的圖像特征；

根據(jù)所述增強(qiáng)后所述圖像特征進(jìn)行圖像重建，得到第一圖像處理數(shù)據(jù)。

在一實(shí)施例中，所述通過顏色矯正網(wǎng)絡(luò)對(duì)轉(zhuǎn)化后的所述水下初始圖像的AB通道數(shù)據(jù)進(jìn)行處理，得到第二圖像處理數(shù)據(jù)的步驟包括：

將所述水下初始圖像的AB通道數(shù)據(jù)輸入顏色矯正網(wǎng)絡(luò)對(duì)應(yīng)的編碼單元，其中，所述編碼單元中的下采樣層對(duì)所述AB通道數(shù)據(jù)進(jìn)行下采樣，得到所述編碼單元的輸出圖像數(shù)據(jù)；

對(duì)所述輸出圖像數(shù)據(jù)進(jìn)行特征提取，得到對(duì)應(yīng)的顏色特征；

將所述顏色特征輸入所述顏色矯正網(wǎng)絡(luò)對(duì)應(yīng)的解碼單元，其中，所述解碼單元中的上采樣塊對(duì)所述顏色特征進(jìn)行上采樣，得到第二圖像處理數(shù)據(jù)。

在一實(shí)施例中，所述根據(jù)所述第一圖像處理數(shù)據(jù)和第二圖像處理數(shù)據(jù)生成增強(qiáng)圖像的步驟包括：