使用python在眼睛图像中进行瞳孔检测

Question

我需要在像这样的眼睛中标记瞳孔。我已经写了这段代码

img_name='6.jpg'
image = cv2.imread(img_name)
image_copy_new=cv2.imread(img_name)
gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
retval, thresholded = cv2.threshold(gray, 30, 255, cv2.THRESH_BINARY_INV)
plt.imshow(thresholded,cmap="gray")

这会产生这样的输出-

然后我在图像中搜索轮廓，并尝试通过此代码仅在图像中找到最圆的

contours, hierarchy = cv2.findContours(thresholded, cv2.RETR_TREE, cv2.CHAIN_APPROX_SIMPLE)
image_copy = np.zeros_like(image)  # create a new emtpy image
for cnt in contours:
    peri = cv2.arcLength(cnt, True)
    approx = cv2.approxPolyDP(cnt, 0.04 * peri, True)
    (x, y, w, h) = cv2.boundingRect(cnt)
    ar = w / float(h)
    if w*h > 20 and 0.9 < ar < 1.1:  # filtering condition
        cv2.drawContours(image, [cnt], 0, 255, -1)

尽管在某些情况下，如果眼睛朝前，这会产生很好的效果，但是在其他情况下（如这种情况），它会完全失败。我尝试了许多其他操作，例如“可变换，不同的变形” ，但我无法解决此问题。 图像只有眼睛，不是整个脸，否则dlibs人脸检测就可以了 此代码有效的情况是

感谢您抽出宝贵时间来帮助我。

Answer 1

添加一些模糊，腐蚀和膨胀可能会有所帮助。腐蚀会去除很小的特征，例如睫毛周围的噪音，而扩张会把所有尚存的点恢复到原来的大小。通过调整腐蚀和膨胀尺寸，您应该能够消除大部分噪声，并使中央瞳孔看起来更好。

以下是我如何执行此操作的示例：

    gray = cv2.cvtColor(frame_in, cv2.COLOR_BGR2GRAY)
    blurred = cv2.GaussianBlur(gray, (5, 5), 0)
    thresh = cv2.threshold(blurred, 30, 255, cv2.THRESH_BINARY)[1]
    erosion_size = 10
    dilate_size = 8
    thresh = cv2.erode(thresh, cv2.getStructuringElement(cv2.MORPH_ELLIPSE, (erosion_size, erosion_size)))
    thresh = cv2.dilate(thresh, cv2.getStructuringElement(cv2.MORPH_ELLIPSE, (dilate_size, dilate_size)))