diff options
author | Patrick Simianer <p@simianer.de> | 2016-11-26 12:40:34 +0100 |
---|---|---|
committer | Patrick Simianer <p@simianer.de> | 2016-11-26 12:40:34 +0100 |
commit | 162187608bbaf1f79d38c88803754c8e58359129 (patch) | |
tree | 24a10ed5f715585215377854af6c0cbde276ca83 /pupil-detect.cpp | |
parent | 89ec4030ba4e426e6e7992336a6665ef9d37ec48 (diff) |
cleanup
Diffstat (limited to 'pupil-detect.cpp')
-rw-r--r-- | pupil-detect.cpp | 52 |
1 files changed, 0 insertions, 52 deletions
diff --git a/pupil-detect.cpp b/pupil-detect.cpp deleted file mode 100644 index 6e51c94..0000000 --- a/pupil-detect.cpp +++ /dev/null @@ -1,52 +0,0 @@ -/** - * Program to detect pupil, based on - * http://www.codeproject.com/Articles/137623/Pupil-or-Eyeball-Detection-and-Extraction-by-C-fro - * with some improvements. - */ -#include <opencv2/imgproc/imgproc.hpp> -#include <opencv2/highgui/highgui.hpp> -#include <cmath> - -int main(int argc, char** argv) -{ - // Load image - cv::Mat src = cv::imread("eye_image.jpg"); - if (src.empty()) - return -1; - - // Invert the source image and convert to grayscale - cv::Mat gray; - cv::cvtColor(~src, gray, CV_BGR2GRAY); - - // Convert to binary image by thresholding it - cv::threshold(gray, gray, 220, 255, cv::THRESH_BINARY); - - // Find all contours - std::vector<std::vector<cv::Point> > contours; - cv::findContours(gray.clone(), contours, CV_RETR_EXTERNAL, CV_CHAIN_APPROX_NONE); - - // Fill holes in each contour - cv::drawContours(gray, contours, -1, CV_RGB(255,255,255), -1); - - for (int i = 0; i < contours.size(); i++) - { - double area = cv::contourArea(contours[i]); - cv::Rect rect = cv::boundingRect(contours[i]); - int radius = rect.width/2; - - // If contour is big enough and has round shape - // Then it is the pupil - if (area >= 30 && - std::abs(1 - ((double)rect.width / (double)rect.height)) <= 0.2 && - std::abs(1 - (area / (CV_PI * std::pow(radius, 2)))) <= 0.2) - { - cv::circle(src, cv::Point(rect.x + radius, rect.y + radius), radius, CV_RGB(255,0,0), 2); - } - } - - cv::imshow("image", src); - cv::waitKey(0); - - return 0; -} - |