基于HSI和局部同态滤波的彩色图像增强

2021年7月6日 17点热度 0条评论 来源: 奈何缘浅

简介

在图像采集过程中,由于光照环境或物体表面反光等原因会造成图像光照不均 。图像的光照不均会直接影响图像分析的结果。因此,对光照不均图像进行增强,消除光照的影响是光照不均图像处理中不可缺少的环节 。同态滤波是光照不均图像增强常用的方法。同态滤波增强是从图像的整体角度对光照不均进行修正,虽然可以很好地保持图像的原始面貌,但它没有充分考虑图像的空间局部特性,在增强图像某部分像素时,易导致另一部分像素过增强 。因此,局部同态滤波算法对图像的各个子图像分别进行同态滤波,并对局部同态滤波产生的块效应采用相邻子图像边界平均的方法进行去除。

项目源码地址:https://github.com/gain-wyj/Color-image-enhancement

欢迎大家star

HSI 颜色模型

HSI 颜色模型是一种直观的颜色模型,它从人眼视觉系统出发,用 H(Hue)、S(Saturation)和 I(Intensity)分别表示色调、饱和度和亮度 3 种独立的颜色特征。HSI 模型的建立基于两个重要的事实:其一,I 分量与图像的彩色信息无关;其二,H 和 S 分量与人感受颜色的方式是一致的。这些特点使得 HSI 模型非常适合借助于人的视觉系统来感知彩色特性的图像处理算法。

色调H(Hue):与光波的频率有关,它表示人的感官对不同颜色的感受,如红色、绿色、蓝色等,它也可表示一定范围的颜色,如暖色、冷色等。

饱和度S(Saturation):表示颜色的纯度,纯光谱色是完全饱和的,加入白光会稀释饱和度。饱和度越大,颜色看起来就会越鲜艳,反之亦然。

亮度I(Intensity):对应成像亮度和图像灰度,是颜色的明亮程度。

若将RGB单位立方体沿主对角线进行投影,可得到六边形,这样,原来沿主对角线的灰色都投影到中心白色点,而红色点(1,0,0)则位于右边的角上,绿色点(0,1,0)位于左上角,蓝色点(0,0,1)则位于左下角。

HSI颜色模型的双六棱锥表示,I是强度轴,色调H的角度范围为[0,2π],其中,纯红色的角度为0,纯绿色的角度为2π/3,纯蓝色的角度为4π/3。饱和度S是颜色空间任一点距I轴的距离。当然,若用圆表示RGB模型的投影,则HSI色度空间为双圆锥3D表示。

注意: 当强度I=0时,色调H、饱和度S无定义;当S=0时,色调H无定义。

RGB 模型转换到 HSI 模型

给的一幅 RGB 彩色格式的图像,对[0,1]范围内的 R、G、B值,其对应的 HSI 模型中的 I、S、H 分量的计算公式分别为:

HSI 模型转换到 RGB 模型

S 和 I 的值在[0,1]之间,R、G、B 的值也在[0,1]之间,HSI模型转换为 RGB 模型的公式分为 3 段。

(1) 当 0° ≤ H < 120° 时:

(2) 当 120° ≤ H < 240° 时:

(3) 当 240° ≤ H < 360° 时:

同态滤波

从图像的形成和光特性考虑,一幅图像是由光源的照度分量 i(x,y) 和目标物的反射分量 r(x,y) 组成的,其数学模型为:f(x,y) = i(x,y)·r(x,y) ,式中,r(x,y) 的性质取决于成像物体的表面特性。

一般来说,光照条件体现在照度分量 i(x,y) 中,i(x,y) 变化缓慢,其频谱主要落在低频区域;而 r(x,y) 反映图像的细节等特征,其频谱主要落在高频部分。为此,只要从 f(x,y) 中将i(x,y)r(x,y) 分开,并分别采取压缩低频、提升高频的方法,就可以达到减弱照度分量、增强反射分量从而使图像清晰的目的 。

同态滤波的流程如图所示。

基本思路

在 HSI 颜色空间下,对 I 分量进行处理,裁剪成 相同大小的 n×n 图像块,进行同态滤波,以此实现局部增强,但拼接图像时,在边缘必然会存在颜色突变,造成块效应,因此需要解决这个副作用。将相邻的图像块分为水平和竖直两类,利用边界处左右两边的像素进行均值滤波,以此来消除块效应。增强后的 I 分量再重新与 S 和 H 分量组合并还原到 RGB 空间。

代码实现

同态滤波实现

对彩色图像进行同态滤波,首先需要将彩色图像从 RGB 颜色空间转换到 HSI 颜色空间,然后保持色调和饱和度分量不变,只对亮度分量进行同态滤波,最后再将处理后的图像从 HSI 空间转换到 RGB 颜色空间,得到增强后的彩色图像。
HomoMor.m

function im_e = HomoMor(im,Hh,Hl,D0,c)
% 高斯同态滤波器参数的设置
% Hh = 1.2;  % 高频增益,需要大于1
% Hl = 0.5;  % 低频增益,取值在0和1之间
% D0 = 4;    % 截止频率,越大图像越亮
% c = 1;     % 锐化系数

%% 滤波器初始化
im = double(im);
[row, col] = size(im);

% 确定傅里叶变换的原点
x0 = floor(row/2);
y0 = floor(col/2);

% 初始化
H = zeros(row,col);

for i = 1:row
    for j = 1:col
        D = (i-x0)^2 + (j-y0)^2;
        if D == 0
            H(i,j) = Hl;
        else
            H(i,j) = (Hh-Hl) * (1 - exp(-c*D^2/(D0^2))) + Hl;  % 高斯同态滤波函数
        end
    end
end

%% 同态滤波
im_l = log(im + 0.000001);              % 取对数变换
im_f = fftshift(fft2(im_l));            % 傅里叶变换,并移到中心位置
im_nf = H .* im_f;                      % 高斯滤波
im_n = real(ifft2(ifftshift(im_nf)));   % 傅里叶反变换,恢复位置
im_e = exp(im_n - 0.000001);            % 取指数变化
end

HSV 颜色空间下的全局同态滤波

GlobalHomo.m

function rgbim = GlobalHomo(I,Hh,Hl,D0,c)
%全局同态滤波
% rgbim = GlobalHomo(I,Hh,Hl,D0,c)
% I:输入图像
% Hh:高频增益,需要大于1
% Hl:低频增益,取值在0和1之间
% D0:截止频率,越大图像越亮
% c:锐化系数

hsiim = rgb2hsi(I);
im = hsiim(:,:,3);
im_g = HomoMor(im,Hh,Hl,D0,c);
hsiim(:,:,3) = im_g;
rgbim = hsi2rgb(hsiim);
end

不加块效应消除的分块同态滤波

BlockHomo.m

function rgbim = BlockHomo(I,Hh,Hl,D0,c,block)
% rgbim = GlobalHomo(I,Hh,Hl,D0,c)
% I:输入图像
% Hh:高频增益,需要大于1
% Hl:低频增益,取值在0和1之间
% D0:截止频率,越大图像越亮
% c:锐化系数
% block:分块大小


hsiim = rgb2hsi(I);
im = hsiim(:,:,3);

[row,col] = size(im);
% 块的大小为Mb*Nb
Mb = block; Nb = block;
rb = floor(row/Mb); cb = floor(col/Nb);
im_g = im;
for i =1:rb
    for j = 1:cb
        temp = im((i-1)*Mb+1:i*Mb , (j-1)*Nb+1:j*Nb);
        im_g((i-1)*Mb+1:i*Mb , (j-1)*Nb+1:j*Nb) = HomoMor(temp,Hh,Hl,D0,c);
    end
end
hsiim(:,:,3) = im_g;
rgbim = hsi2rgb(hsiim);

end

加去除块效应的分块同态滤波

BlockHomo.m

function rgbim = BlockHomo(I,Hh,Hl,D0,c,block)
%分块同态滤波
% rgbim = GlobalHomo(I,Hh,Hl,D0,c)
% I:输入图像
% Hh:高频增益,需要大于1
% Hl:低频增益,取值在0和1之间
% D0:截止频率,越大图像越亮
% c:锐化系数
% block:分块大小


hsiim = rgb2hsi(I);
im = hsiim(:,:,3);

[row,col] = size(im);
% 块的大小为Mb*Nb
Mb = block; Nb = block;
rb = floor(row/Mb); cb = floor(col/Nb);
im_g = im;
for i =1:rb
    for j = 1:cb
        temp = im((i-1)*Mb+1:i*Mb , (j-1)*Nb+1:j*Nb);
        im_g((i-1)*Mb+1:i*Mb , (j-1)*Nb+1:j*Nb) = HomoMor(temp,Hh,Hl,D0,c);
    end
end

% 分块同态滤波 
M = 3; N = 3;
n0 = floor(N/2); m0 = floor(M/2);
Ide = im_g;

% 去除水平相邻子图像块效应
for i = 1:row
    for j = Nb:Nb:col-Nb
        temp1 = 0;
        temp2 = 0;
        for k = -n0:n0
            temp1 = temp1 + im_g(i,j+k);
            temp2 = temp2 + im_g(i,j+1+k);
        end
        Ide(i,j) = temp1/N;
        Ide(i,j+1) = temp2/N;
    end
end
% 去除垂直相邻子图像块效应
for i = Mb:Mb:row-Nb
    for j =  1:col
        temp1 = 0;
        temp2 = 0;
        for k = -m0:m0
            temp1 = temp1 + im_g(i+k,j);
            temp2 = temp2 + im_g(i+1+k,j);
        end
        Ide(i,j) = temp1/M;
        Ide(i+1,j) = temp2/M;
    end
end
hsiim(:,:,3) = Ide;
rgbim = hsi2rgb(hsiim);

end

局部平均信息熵

BlockAvEn.m

function en = BlockAvEn(I,block)
%局部平均信息熵
% en = BlockAvEn(I,block)
% I:输入图像
% block:分块大小
en = 0;
[row,col,channel] = size(I);
% 块的大小为Mb*Nb
Mb = block; Nb = block;
rb = floor(row/Mb); cb = floor(col/Nb);
for c = 1:channel
    im = I(:,:,c);
    for i =1:rb
        for j = 1:cb
            temp = im((i-1)*Mb+1:i*Mb , (j-1)*Nb+1:j*Nb);
            en = en + entropy(temp);
        end
    end
end
en = en/(rb*cb*channel);
end


实验结果对比

fig.m

close all;clear;clc
addpath('model');
addpath('src');


%  读取图像
I = imread('4.1.01.tiff');
en = BlockAvEn(I,8);

% 全局同态滤波
rgbim0 =  GlobalHomo(I,1.2,0.5,50,1);
en0 = BlockAvEn(rgbim0,8);

% 局部同态滤波
rgbim1 = LocalHomo(I,1.2,0.5,4,1,8);
en1 = BlockAvEn(rgbim1,8);

% 分块同态滤波 
rgbim2 = BlockHomo(I,1.2,0.5,4,1,8);
en2 = BlockAvEn(rgbim2,8);

figure(1)
subplot(1,4,1)
imshow(I);
title(['原始图像:',num2str(en)])

subplot(1,4,2)
imshow(rgbim0);
title(['全局同态滤波:',num2str(en0)])

subplot(1,4,3)
imshow(rgbim1);
title(['局部同态滤波:',num2str(en1)])

subplot(1,4,4)
imshow(rgbim2);
title(['分块同态滤波:',num2str(en2)])

结果

图像 原文的局部平均信息熵 复现的局部平均信息熵
原始图像 3.8476 3.8476
全局同态滤波 4.6636 4.6635
局部同态滤波 4.6848 4.6848
去除块效应的局部同态滤波 4.6740 4.6758

从实验结果可以看出,全局同态滤波在增强图像某部分像素时,导致另一部分像素过增强,从而造成图像细节的损失。而局部同态滤波考虑了图像的局部特征,增强之后的图像光照均匀,明暗适中,细节清晰,且经过块效应去除后,图像有更好的视觉效果。

    原文作者:奈何缘浅
    原文地址: https://www.cnblogs.com/nhyq-wyj/p/14974507.html
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系管理员进行删除。