消除红眼效应及matlab实现

最新推荐文章于 2023-06-16 18:59:13 发布

rrr2

最新推荐文章于 2023-06-16 18:59:13 发布

阅读量5.6k

点赞数 6

分类专栏： matlab 文章标签： matlab 红眼消除

本文链接：https://blog.csdn.net/qq_35608277/article/details/78363139

版权

matlab 专栏收录该内容

35 篇文章 13 订阅

订阅专栏

红眼效应:是指用闪光灯拍摄人物照片时,在人眼瞳孔中央形成的红眼现象，有碍观瞻。
* *
消除红眼思路：建立红眼模型->提取红眼区域->进行红眼修正->图像输出。
具体实现过程：
1.HSI模型：HSI模型是美国色彩学家蒙塞尔于1915年提出的，它反映了人的视觉系统感知彩色的方式，以色调、饱和度和强度三种基本特征量来感知颜色。饱和度S(Saturatuin)表示颜色的纯度，饱和度越大，颜色看起来就会越鲜艳，反之亦然。强度I(Intensity)，对应成像亮度和图像灰度，是颜色的明亮程度。
RGB与HSI转换关系有公式：
这里写图片描述

2.红眼模型：
这里用HSI模型建立红眼模型。

这里写图片描述

matlab实现：
首先保存两个图片类型转换函数，注意两个m文件名与函数名相同。

RGB到HSI：

function [hsi,H,S,I]=rgb2hsi(rgb)
% RGB转换到HSI
rgb=im2double(rgb);
r=rgb(:,:,1);
g=rgb(:,:,2);
b=rgb(:,:,3);
%Implement the conversion equations.
num=0.5*((r-g)+(r-b));
den=sqrt((r-g).^2+(r-b).*(g-b));
theta=acos(num./(den+eps));
H=theta;
H(b>g)=2*pi-H(b>g);
H=H/(2*pi);

num=min(min(r,g),b);
den=r+g+b;
den(den==0)=eps;
S=1-3.*num./den;
H(S==0)=0;
I=(r+g+b)/3;
%Combine all three results into an hsi image.
hsi=cat(3,H,S,I);

HSI到RGB：

function rgb=hsi2rgb(hsi)
% HSI转换到RGB
H=hsi(:,:,1)*2*pi;
S=hsi(:,:,2);
I=hsi(:,:,3);

%Implement the conversion equations.
R=zeros(size(hsi,1),size(hsi,2));
G=zeros(size(hsi,1),size(hsi,2));
B=zeros(size(hsi,1),size(hsi,2));

% RG sector (0<=H<2*pi/3).
idx=find((0<=H)&(H<2*pi/3));
B(idx)=I(idx).*(1-S(idx));
R(idx)=I(idx).*(1+S(idx).*cos(H(idx))./cos(pi/3-H(idx)));
G(idx)=3*I(idx)-(R(idx)+B(idx));
%BG sector (2*pi/3<=H<4*pi/3).
idx=find((2*pi/3<=H)&(H<4*pi/3));
R(idx)=I(idx).*(1-S(idx));
G(idx)=I(idx).*(1+S(idx).*cos(H(idx)-2*pi/3)./cos(pi-H(idx)));
B(idx)=3*I(idx)-(R(idx)+G(idx));
%BR sector.
idx=find((4*pi/3<=H)&(H<=2*pi));
G(idx)=I(idx).*(1-S(idx));
B(idx)=I(idx).*(1+S(idx).*cos(H(idx)-4*pi/3)./cos(5*pi/3-H(idx)));
R(idx)=3*I(idx)-(G(idx)+B(idx));
rgb=cat(3,R,G,B);
rgb=max(min(rgb,1),0);

主程序：

%本程序的功能是去除用户指定区域的红眼
clear all;clc;%内存清理
[file, path] = uigetfile('*.bmp', 'Image');%获取图像名及存储位置
if isequal(file,0)                         %未选择图像，退出
    disp('取消选择');
else
    file=[path,file];                      %获取图像路径
    [filename,map]=imread(file);           %读取图像
    if  ndims(filename)~=3                 %是否为灰度图像,是退出并显示图像
        disp('非彩色图像');
        imshow(filename);
    else                                   %去红眼操作
       goagain=1;                  %操作控制变量，1进行操作
       change=2;           %参数更改控制变量，1更改预设参数
       h1=1/6;      %预设对rol区域内-pi/3<H<pi/3,S>0.2的像素点进行去红眼操作
       h2=1/6;
       s=0.2;
       fprintf('预设对rol区域内-pi/3<H<pi/3,S>0.2的像素点进行去红眼操作，即令S=0\n')
       while goagain==1                           
           fprintf('是否修改H、S预设值：1、是；2、否\n')
           change=input('请选择：');        %是否更改预设参数
           while change~=1&change~=2         
               fprintf('错误命令！请重新输入~~\n');
               change=input('请选择：');
           end
           if change==1                        %预设参数更改
               h1=input('请输入H下限值（-2*pi/3~0）：')/pi/2;
               while (h1>0)||(h1<-1/3)
                   fprintf('错误命令！请重新输入~~\n');
                   h1=input('请输入H下限值（-2*pi/3~0）：')/pi/2;
               end;
               h2=input('请输入H上限值（0~2*pi/3）：')/pi/2;
               while (h2<0)||(h2>1/3)
                   fprintf('错误命令！请重新输入~~\n');
                   h2=input('请输入H上限值（0~2*pi/3）：')/pi/2
               end;
               s=input('请输入S值(0~1)：');
               while (s>1)||(s<0)
                   fprintf('错误命令！请重新输入~~\n');
                   s=input('请输入S值(0~1)：');
               end;
               change=2;
           end;
           B=roipoly(filename);            %绘制要去红眼的rol区域，返回与filename大小相同的逻辑矩阵B，rol内像素值为1
                                           %用户通过鼠标绘制任意多边形，点左键绘制顶点，点右键绘制最后的一
                                           %个顶点
          [hsi,H,S,I]=rgb2hsi(filename);  %rgb->hsi




           rol=find(B~=0);                 %获取rol内像素数
           [L,M]=size(rol);
           for i=1:L                       %遍历rol内像素，判定-1/6<h<1/6且s>0.2为红眼，并令s=0以去除红眼
               if (((H(rol(i))>0)&(H(rol(i))<h2))|((H(rol(i))>(1-abs(h1)))&(H(rol(i))<1)))
                   if S(rol(i))>s
                       S(rol(i))=0;
                       I(rol(i))=0;
                   end;
               end;
           end;
           hsi=cat(3,H,S,I);            %生成去红眼后的hsi图像
           rgb=hsi2rgb(hsi);               %hsi->rgb
           imshow(rgb);              %显示去红眼后的rgb图像
           fprintf('是否继续操作：1、是；2、否\n');%提示是否继续操作
           goagain=input('请选择：');
           while goagain~=1&goagain~=2
              fprintf('错误命令！请重新输入~~\n');
              goagain=input('请选择：');
           end
           if goagain==1                   
               filename=rgb;
               clf;
           end;
       end;
    end
end;