Last active
March 16, 2018 11:31
-
-
Save zzuummaa/64e3178dc13d57a9748bc23426a22ce6 to your computer and use it in GitHub Desktop.
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
package ru.zuma; | |
import java.util.ArrayDeque; | |
import static org.bytedeco.javacpp.opencv_core.*; | |
import static org.bytedeco.javacpp.opencv_objdetect.*; | |
public class AsyncClassifier extends Thread { | |
interface DetectionCallback { | |
void onDetections(RectVector detections); | |
} | |
private CascadeClassifier classifier; | |
private RectVector detections; | |
private ArrayDeque<Mat> imageQueue; | |
private DetectionCallback detectionCallback; | |
public AsyncClassifier(CascadeClassifier classifier) { | |
this(classifier, true); | |
} | |
public AsyncClassifier(CascadeClassifier classifier, boolean isAutoStart) { | |
this.classifier = classifier; | |
this.detections = new RectVector(); | |
this.imageQueue = new ArrayDeque<Mat>(1); | |
this.detectionCallback = null; | |
if (isAutoStart) { | |
start(); | |
} | |
} | |
public synchronized void setOnDetections(DetectionCallback detectionCallback) { | |
this.detectionCallback = detectionCallback; | |
} | |
public RectVector getDetections() { | |
return new RectVector(detections); | |
} | |
public void setImage(Mat image) { | |
synchronized (imageQueue) { | |
if (!imageQueue.isEmpty()) { | |
imageQueue.clear(); | |
} | |
imageQueue.push(image); | |
imageQueue.notify(); | |
} | |
} | |
public void detect(Mat image, RectVector detections) { | |
classifier.detectMultiScale(image, detections); // Performs the detection | |
} | |
@Override | |
public void run() { | |
Mat image; | |
while (!isInterrupted()) { | |
synchronized (imageQueue) { | |
try { | |
while (imageQueue.isEmpty()) { | |
imageQueue.wait(); | |
} | |
} catch (InterruptedException e) { | |
break; | |
} | |
image = imageQueue.pop(); | |
} | |
detect(image, detections); | |
synchronized (this) { | |
if (detectionCallback != null) { | |
detectionCallback.onDetections(detections); | |
} | |
} | |
} | |
} | |
} |
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
package ru.zuma; | |
import static org.bytedeco.javacpp.opencv_core.*; | |
public class Main { | |
public static void main(String[] args) { | |
Vision vision = new Vision(); | |
final RectVector[] diceDetections = new RectVector[1]; | |
diceDetections[0] = new RectVector(); | |
AsyncClassifier classifier = vision.getClassifier(); | |
classifier.setOnDetections( (RectVector detections) -> diceDetections[0] = detections ); | |
} | |
} | |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment