我在做瞳孔检测为我的学校项目。这是我第一次使用Python 3.4.2和OpenCV 3.1.0与OpenCV和Python合作。在OpenCV和Python中的瞳孔检测
我正在使用Raspberry Pi NoIR相机,并且我获得了很好的图像。 。
但我不能很好地检测出瞳孔(因为闪烁,睫毛和阴影 我指的是网络上的一些代码,以下是部分代码
...
kernel = cv2.getStructuringElement(cv2.MORPH_RECT, (5, 5))
# capture frames from the camera
for frame in camera.capture_continuous(rawCapture, format="bgr", use_video_port=True):
image = frame.array
cv2.imshow("image", image)
gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
retval, thresholded = cv2.threshold(gray, 80, 255, 0)
cv2.imshow("threshold", thresholded)
closed = cv2.erode(cv2.dilate(thresholded, kernel, iterations=1), kernel, iterations=1)
#closed = cv2.morphologyEx(close, cv2.MORPH_CLOSE, kernel)
cv2.imshow("closed", closed)
thresholded, contours, hierarchy = cv2.findContours(closed, cv2.RETR_LIST, cv2.CHAIN_APPROX_NONE)
drawing = np.copy(image)
cv2.drawContours(drawing, contours, -1, (255, 0, 0), 2)
for contour in contours:
area = cv2.contourArea(contour)
bounding_box = cv2.boundingRect(contour)
extend = area/(bounding_box[2] * bounding_box[3])
# reject the contours with big extend
if extend > 0.8:
continue
# calculate countour center and draw a dot there
m = cv2.moments(contour)
if m['m00'] != 0:
center = (int(m['m10']/m['m00']), int(m['m01']/m['m00']))
cv2.circle(drawing, center, 3, (0, 255, 0), -1)
# fit an ellipse around the contour and draw it into the image
try:
ellipse = cv2.fitEllipse(contour)
cv2.ellipse(drawing, box=ellipse, color=(0, 255, 0))
except:
pass
# show the frame
cv2.imshow("Drawing", drawing)
...
输入图像:
输出图像:
如何删除与瞳孔无关的图像部分,如上所示?
除了答案,任何提示也欢迎。
相关:[加快在numpy的量化眼睛跟踪算法(https://stackoverflow.com/questions/35996257/speeding-up-vectorized-eye-tracking-algorithm-in-numpy) 。您也可以检查循环性([示例代码](https://github.com/Itseez/opencv/blob/3.1.0/modules/features2d/src/blobdetector.cpp#L222))。 – Catree
其他选项:用[HoughCircles]直接检测(http://docs.opencv.org/3.1.0/dd/d1a/group__imgproc__feature.html#ga47849c3be0d0406ad3ca45db65a25d2d)和/或只保留轮廓,如果里面的区域比外。如果眼睛始终居中且距离相同,则还可以定义感兴趣区域(ROI)+使用该区域。 – Catree
我会在二进制映像上使用[erode](http://docs.opencv.org/2.4/modules/imgproc/doc/filtering.html?highlight=erode#erode),然后简单地执行[HoughCircles](http: //docs.opencv.org/2.4/doc/tutorials/imgproc/imgtrans/hough_circle/hough_circle.html)来检测图像中最重要的圆。 – 0x90