by.zuyeu.deyestracker.core.detection.task.DetectPupilsTask.java Source code

Introduction

Here is the source code for by.zuyeu.deyestracker.core.detection.task.DetectPupilsTask.java

Source

/*
 * To change this license header, choose License Headers in Project Properties.
 * To change this template file, choose Tools | Templates
 * and open the template in the editor.
 */
package by.zuyeu.deyestracker.core.detection.task;

import org.opencv.core.Core;
import org.opencv.core.Mat;
import org.opencv.core.Point;
import org.opencv.core.Size;
import org.opencv.imgproc.Imgproc;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

/**
 *
 * @author Fieryphoenix
 */
public class DetectPupilsTask implements IDetectTask<Point> {

    private static final Logger LOG = LoggerFactory.getLogger(DetectPupilsTask.class);
    private static final int GAUS_BLUR_DELTA = 2;

    private static final Size STRUCT_ELEMENT_SIZE = new Size(15, 15);
    private final Mat frame;

    public DetectPupilsTask(final Mat frame) {
        this.frame = frame;
    }

    @Override
    public Point call() throws Exception {
        long startTime = System.nanoTime();

        final Mat imageHSV = new Mat(frame.size(), Core.DEPTH_MASK_8U);
        Imgproc.cvtColor(frame, imageHSV, Imgproc.COLOR_BGR2GRAY);

        Imgproc.erode(imageHSV, imageHSV, Imgproc.getStructuringElement(Imgproc.MORPH_RECT, STRUCT_ELEMENT_SIZE));
        Imgproc.dilate(imageHSV, imageHSV, Imgproc.getStructuringElement(Imgproc.MORPH_RECT, STRUCT_ELEMENT_SIZE));
        Imgproc.GaussianBlur(imageHSV, imageHSV, STRUCT_ELEMENT_SIZE, GAUS_BLUR_DELTA);

        Core.MinMaxLocResult mmG = Core.minMaxLoc(imageHSV);

        long endTime = System.nanoTime();
        LOG.debug("pupil detected = {}", mmG.minLoc);
        LOG.debug("detection time: {} ms", (float) (endTime - startTime) / 1000000);
        return mmG.minLoc;
    }
}