在本教程中,我们将学习如何完全自动消除照片中的红眼。

 

仅仅考虑消除红眼的问题,就会带回我小时候的记忆。小时候拥有一台傻瓜相机“ Hotshot”。它可能是我们家庭中唯一的便携式电子设备,使用了数十年。我妈妈把相机和她的黄金首饰锁在一个保险箱中。摄影是为非常特殊的场合而保留的,因为胶卷很贵。有时要过几个月才能拍摄出我们拍摄的照片。自然,当我们晚上拍摄的照片产生红眼效果时,它总是让我心碎。满脸鲜血的微笑的人使我想起了德古拉!

 

在本科期间的某个时候,我想出了一种使用照片编辑工具手动消除红眼的方法。学习基本原理给我带来了很多快乐,它的简单性让我感到惊讶!我要花几年的时间才能弄清楚如何完全自动地做到这一点。

 

构建一个适用于各种图像的健壮的红眼消除应用程序超出了本文的范围。但是,我们将学习基本原理并构建概念证明应用程序。

 

是什么导致闪光灯摄影中出现红眼效应?

opencv 摄像机调整曝光 opencv关闭摄像头自动曝光_图像算法

 

opencv 摄像机调整曝光 opencv关闭摄像头自动曝光_图像算法_02

当您在黑暗的房间里时,您的瞳孔会变大(放大),以便让更多的光线进入,以帮助您更好地看到。大多数相机上的闪光灯都非常靠近镜头。当您在闪光灯开启的情况下拍照时,闪光灯发出的光线会通过扩大的瞳孔到达眼球后部,然后再通过瞳孔射入相机的镜头。眼球的背面称为眼底。它是红色的,因为眼底有充足的血液供应。

 

眼底图像显示在左侧。检查眼底可以揭示一个人的健康状况。您甚至可以获取智能手机应用程序,以帮助您查看带有附件的眼底。

 

这些天大多数相机闪光灯会闪烁几秒钟,这会使瞳孔收缩,从而减少了红眼的可能性。

 

如何自动删除红眼?

在本节中,我们将逐步介绍用于自动消除红眼的算法。

 

步骤1:眼睛检测

OpenCV眼睛检测

opencv 摄像机调整曝光 opencv关闭摄像头自动曝光_瞳孔检测_03

第一步是自动检测眼睛。我们使用标准的OpenCV Haar检测器(haarcascade_eye.xml)来寻找眼睛。有时先运行面部检测器,然后再检测面部区域内的眼睛是有意义的。为了简单起见,我们直接在图像上运行眼睛检测器。当输入图像为人像拍摄或眼睛特写时,可以跳过脸部检测器。

 

您还可以说明训练自己的HAAR对象检测器。

 

我在下面共享用于眼睛检测器的代码。

 

C++

// Read image
Mat img = imread("red_eyes.jpg",CV_LOAD_IMAGE_COLOR);
 
// Output image
Mat imgOut = img.clone();
     
// Load HAAR cascade 
CascadeClassifier eyes_cascade("haarcascade_eye.xml");
 
// A vector of Rect for storing bounding boxes for eyes.
std::vector<Rect> eyes;
 
// Detect eyes. 
eyesCascade.detectMultiScale( img, eyes, 1.3, 4, 0 | CASCADE_SCALE_IMAGE, Size(100, 100) );

Python

# Read image
img = cv2.imread("red_eyes.jpg", cv2.IMREAD_COLOR)
 
# Output image
imgOut = img.copy()
     
# Load HAAR cascade
eyesCascade = cv2.CascadeClassifier("haarcascade_eye.xml")
 
# Detect eyes
eyes = eyesCascade.detectMultiScale(img,scaleFactor=1.3, minNeighbors=4, minSize=(100, 100))

步骤2:遮住红眼

接下来,我们需要找到受红眼影响的瞳孔部分。找到红色的方法有很多。需要注意的一件事是我们的颜色不仅是红色,还是鲜红色!您可以根据色相和亮度将图像转换为HSV颜色空间和阈值。在本文中,我们使用了一种更简单的启发式方法。我们说他的红色通道应该大于阈值,也应该大于绿色和蓝色通道的总和。出于概念验证系统的目的,试探法就足够了,但是如果您要为商业软件包构建自动防红眼功能,则需要收集数千张红眼图像才能提出更好的解决方案。

 

在下面的代码中,我们遍历在上一步中检测到的所有眼睛矩形。然后,我们使用命令split将彩色图像分成三个通道。最后,对于红色通道高于阈值(150)且红色通道大于绿色通道和蓝色通道之和的每个像素,我们创建一个掩码为1的蒙版。

C++

for( size_t i = 0; i < eyes.size(); i++ )
{
     
  // Extract eye from the image.
  Mat eye = img(eyes[i]);
         
  // Split eye image into 3 channels.
  vector<Mat>bgr(3);
  split(eye,bgr);
 
  // Simple red eye detector
  Mat mask = (bgr[2] > 150) & (bgr[2] > ( bgr[1] + bgr[0] ));
}

Python

for (x, y, w, h) in eyes:
   
  # Extract eye from the image.
  eye = img[y:y+h, x:x+w]
   
  # Split eye image into 3 channels
  b = eye[:, :, 0]
  g = eye[:, :, 1]
  r = eye[:, :, 2]
 
  # Add the green and blue channels. 
  bg = cv2.add(b, g)
 
  # Simple red eye detector
  mask = (r > 150) &  (r > bg)
   
  # Convert the mask to uint8 format. 
  mask = mask.astype(np.uint8)*255

步骤3:清理瞳孔面罩

opencv 摄像机调整曝光 opencv关闭摄像头自动曝光_红眼_04

 

 

在上一步中创建的蒙版很可能会带有孔。上图中的左图显示了使用颜色处理获得的原始蒙版。

C++

 

void fillHoles(Mat &mask)
{
     
  Mat mask_floodfill = mask.clone();
  floodFill(mask_floodfill, cv::Point(0,0), Scalar(255));
  Mat mask2;
  bitwise_not(mask_floodfill, mask2);
  mask = (mask2 | mask);
 
}

Python

def fillHoles(mask):
    maskFloodfill = mask.copy()
    h, w = maskFloodfill.shape[:2]
    maskTemp = np.zeros((h+2, w+2), np.uint8)
    cv2.floodFill(maskFloodfill, maskTemp, (0, 0), 255)
    mask2 = cv2.bitwise_not(maskFloodfill)
    return mask2 | mask

另外,最好扩大遮罩,使其覆盖比所需区域稍大的区域。这是因为在边界处,颜色逐渐消失,并且在我们的原始蒙版中可能未捕获到某些红色。在上图中,右图是蒙版的放大图。我们使用下面共享的代码生成扩展的蒙版。

C++

// Clean up mask by filling holes and dilating
fillHoles(mask);
dilate(mask, mask, Mat(), Point(-1, -1), 3, 1, 1);

Python

# Clean up mask by filling holes and dilating
mask = fillHoles(mask)
mask = cv2.dilate(mask, None, anchor=(-1, -1), iterations=3, borderType=1, borderValue=1)

第4步:修复红眼

现在我们有了一个只包含每只眼睛红色区域的蒙版。接下来,我们展示如何处理此蒙版中的区域以修复红眼。

 

我们知道,红眼睛会使图像中的红色通道饱和。换句话说,红色通道中的所有信息均被破坏。我们如何才能找回一些信息?修复红眼时,我们无需检索红色通道中的真实基础纹理;我们只需要找到合理的纹理即可。

 

幸运的是,红眼效果只会破坏红色通道中的纹理;蓝色和绿色通道仍然不错。您可以在下图中看到该位置,其中显示了图像的红色,绿色和蓝色通道。

 

opencv 摄像机调整曝光 opencv关闭摄像头自动曝光_opencv 摄像机调整曝光_05

 

绿色和蓝色通道的组合可用于提出合理的红色通道。例如,我们可以创建一个红色通道,该通道是图像中绿色和蓝色通道的平均值。但是,这样做可能会使瞳孔略带淡淡,看起来不错,但效果不佳。注意中间图像中的紫色调。

 

opencv 摄像机调整曝光 opencv关闭摄像头自动曝光_opencv 摄像机调整曝光_06

 

这给我们带来了一个重要的问题。瞳孔应该是什么颜色?瞳孔是眼睛的开口,眼睛的内部完全是黑暗的。因此,瞳孔应该是无色(灰度)和深色的。代替仅替换瞳孔区域中的红色通道,我们将所有通道替换为绿色和蓝色通道的均值。这消除了紫色调。

 

下面的代码首先通过平均绿色和蓝色通道来创建平均通道。然后,它将所有通道的被遮罩区域内的所有像素替换为该平均通道。

C++

// Calculate the mean channel by averaging
// the green and blue channels
Mat mean = (bgr[0]+bgr[1])/2;
 
// Copy the mean image to blue channel with mask.
mean.copyTo(bgr[0], mask);
 
// Copy the mean image to green channel with mask.
mean.copyTo(bgr[1], mask);
 
// Copy the mean image to red channel with mask.
mean.copyTo(bgr[2], mask);

Python

# Calculate the mean channel by averaging
# the green and blue channels. Recall, bg = cv2.add(b, g)
mean = bg / 2
mask = mask.astype(np.bool)[:, :, np.newaxis]
mean = mean[:, :, np.newaxis]
 
# Copy the eye from the original image. 
eyeOut = eye.copy()
 
# Copy the mean image to the output image. 
np.copyto(eyeOut, mean, where=mask)

步骤5:更换固定眼部区域

在上一步中,我们修复了三个渠道。最后一步是合并三个通道以创建我们的RGB图像,然后将此固定眼图区域放回原始图像中。

 

C++

// Merge the three channels
Mat eyeOut;
merge(bgr,eyeOut);
         
// Copy the fixed eye to the output image. 
eyeOut.copyTo(imgOut(eyes[i]));

Python

# Copy the fixed eye to the output image. 
imgOut[y:y+h, x:x+w, :] = eyeOut

自动去除红眼结果

我们首先在示例图像上显示结果。

 

opencv 摄像机调整曝光 opencv关闭摄像头自动曝光_瞳孔检测_07

请注意,从瞳孔区域删除所有颜色会使图像看起来很漂亮,因为眼睛中心的点是完全白色的。还要注意,在瞳孔的边界上,红色逐渐减弱,但是由于进行了扩张操作,我们仍然捕获了该区域。

 

接下来,我们在眼睛的特写照片上显示结果。如果我们使用了面部检测器,它将不会检测到面部,并且自动眼睛检测器将无法工作。

 

 

opencv 摄像机调整曝光 opencv关闭摄像头自动曝光_红眼_08