mirror of https://github.com/opencv/opencv.git
Andrey Kamaev
13 years ago
parent
0bd287a8f2
commit
50c2443691
11 changed files with 382 additions and 295 deletions
@ -0,0 +1,119 @@ |
||||
package org.opencv.samples.colorblobdetect; |
||||
|
||||
import java.util.ArrayList; |
||||
import java.util.Iterator; |
||||
import java.util.List; |
||||
|
||||
import org.opencv.core.Core; |
||||
import org.opencv.core.CvType; |
||||
import org.opencv.core.Mat; |
||||
import org.opencv.core.MatOfPoint; |
||||
import org.opencv.core.Scalar; |
||||
import org.opencv.imgproc.Imgproc; |
||||
|
||||
public class ColorBlobDetector |
||||
{ |
||||
public void setColorRadius(Scalar radius) |
||||
{ |
||||
mColorRadius = radius; |
||||
} |
||||
|
||||
public void setHsvColor(Scalar hsvColor) |
||||
{ |
||||
double minH = (hsvColor.val[0] >= mColorRadius.val[0]) ? hsvColor.val[0]-mColorRadius.val[0] : 0; |
||||
double maxH = (hsvColor.val[0]+mColorRadius.val[0] <= 255) ? hsvColor.val[0]+mColorRadius.val[0] : 255; |
||||
|
||||
mLowerBound.val[0] = minH; |
||||
mUpperBound.val[0] = maxH; |
||||
|
||||
mLowerBound.val[1] = hsvColor.val[1] - mColorRadius.val[1]; |
||||
mUpperBound.val[1] = hsvColor.val[1] + mColorRadius.val[1]; |
||||
|
||||
mLowerBound.val[2] = hsvColor.val[2] - mColorRadius.val[2]; |
||||
mUpperBound.val[2] = hsvColor.val[2] + mColorRadius.val[2]; |
||||
|
||||
mLowerBound.val[3] = 0; |
||||
mUpperBound.val[3] = 255; |
||||
|
||||
Mat spectrumHsv = new Mat(1, (int)(maxH-minH), CvType.CV_8UC3); |
||||
|
||||
for (int j = 0; j < maxH-minH; j++) |
||||
{ |
||||
byte[] tmp = {(byte)(minH+j), (byte)255, (byte)255}; |
||||
spectrumHsv.put(0, j, tmp); |
||||
} |
||||
|
||||
Imgproc.cvtColor(spectrumHsv, mSpectrum, Imgproc.COLOR_HSV2RGB_FULL, 4); |
||||
|
||||
} |
||||
|
||||
public Mat getSpectrum() |
||||
{ |
||||
return mSpectrum; |
||||
} |
||||
|
||||
public void setMinContourArea(double area) |
||||
{ |
||||
mMinContourArea = area; |
||||
} |
||||
|
||||
public void process(Mat rgbaImage) |
||||
{ |
||||
Mat pyrDownMat = new Mat(); |
||||
|
||||
Imgproc.pyrDown(rgbaImage, pyrDownMat); |
||||
Imgproc.pyrDown(pyrDownMat, pyrDownMat); |
||||
|
||||
Mat hsvMat = new Mat(); |
||||
Imgproc.cvtColor(pyrDownMat, hsvMat, Imgproc.COLOR_RGB2HSV_FULL); |
||||
|
||||
Mat Mask = new Mat(); |
||||
Core.inRange(hsvMat, mLowerBound, mUpperBound, Mask); |
||||
Mat dilatedMask = new Mat(); |
||||
Imgproc.dilate(Mask, dilatedMask, new Mat()); |
||||
|
||||
List<MatOfPoint> contours = new ArrayList<MatOfPoint>(); |
||||
Mat hierarchy = new Mat(); |
||||
|
||||
Imgproc.findContours(dilatedMask, contours, hierarchy, Imgproc.RETR_EXTERNAL, Imgproc.CHAIN_APPROX_SIMPLE); |
||||
|
||||
// Find max contour area
|
||||
double maxArea = 0; |
||||
Iterator<MatOfPoint> each = contours.iterator(); |
||||
while (each.hasNext()) |
||||
{ |
||||
MatOfPoint wrapper = each.next(); |
||||
double area = Imgproc.contourArea(wrapper); |
||||
if (area > maxArea) |
||||
maxArea = area; |
||||
} |
||||
|
||||
// Filter contours by area and resize to fit the original image size
|
||||
mContours.clear(); |
||||
each = contours.iterator(); |
||||
while (each.hasNext()) |
||||
{ |
||||
MatOfPoint contour = each.next(); |
||||
if (Imgproc.contourArea(contour) > mMinContourArea*maxArea) |
||||
{ |
||||
Core.multiply(contour, new Scalar(4,4), contour); |
||||
mContours.add(contour); |
||||
} |
||||
} |
||||
} |
||||
|
||||
public List<MatOfPoint> getContours() |
||||
{ |
||||
return mContours; |
||||
} |
||||
|
||||
// Lower and Upper bounds for range checking in HSV color space
|
||||
private Scalar mLowerBound = new Scalar(0); |
||||
private Scalar mUpperBound = new Scalar(0); |
||||
// Minimum contour area in percent for contours filtering
|
||||
private static double mMinContourArea = 0.1; |
||||
// Color radius for range checking in HSV color space
|
||||
private Scalar mColorRadius = new Scalar(25,50,50,0); |
||||
private Mat mSpectrum = new Mat(); |
||||
private List<MatOfPoint> mContours = new ArrayList<MatOfPoint>();; |
||||
} |
||||
@ -1,61 +0,0 @@ |
||||
/* DO NOT EDIT THIS FILE - it is machine generated */ |
||||
#include <jni.h> |
||||
/* Header for class org_opencv_samples_fd_DetectionBaseTracker */ |
||||
|
||||
#ifndef _Included_org_opencv_samples_fd_DetectionBaseTracker |
||||
#define _Included_org_opencv_samples_fd_DetectionBaseTracker |
||||
#ifdef __cplusplus |
||||
extern "C" { |
||||
#endif |
||||
/*
|
||||
* Class: org_opencv_samples_fd_DetectionBaseTracker |
||||
* Method: nativeCreateObject |
||||
* Signature: (Ljava/lang/String;F)J |
||||
*/ |
||||
JNIEXPORT jlong JNICALL Java_org_opencv_samples_fd_DetectionBaseTracker_nativeCreateObject |
||||
(JNIEnv *, jclass, jstring, jint); |
||||
|
||||
/*
|
||||
* Class: org_opencv_samples_fd_DetectionBaseTracker |
||||
* Method: nativeDestroyObject |
||||
* Signature: (J)V |
||||
*/ |
||||
JNIEXPORT void JNICALL Java_org_opencv_samples_fd_DetectionBaseTracker_nativeDestroyObject |
||||
(JNIEnv *, jclass, jlong); |
||||
|
||||
/*
|
||||
* Class: org_opencv_samples_fd_DetectionBaseTracker |
||||
* Method: nativeStart |
||||
* Signature: (J)V |
||||
*/ |
||||
JNIEXPORT void JNICALL Java_org_opencv_samples_fd_DetectionBaseTracker_nativeStart |
||||
(JNIEnv *, jclass, jlong); |
||||
|
||||
/*
|
||||
* Class: org_opencv_samples_fd_DetectionBaseTracker |
||||
* Method: nativeStop |
||||
* Signature: (J)V |
||||
*/ |
||||
JNIEXPORT void JNICALL Java_org_opencv_samples_fd_DetectionBaseTracker_nativeStop |
||||
(JNIEnv *, jclass, jlong); |
||||
|
||||
/*
|
||||
* Class: org_opencv_samples_fd_DetectionBaseTracker |
||||
* Method: nativeSetFaceSize |
||||
* Signature: (JI)V |
||||
*/ |
||||
JNIEXPORT void JNICALL Java_org_opencv_samples_fd_DetectionBaseTracker_nativeSetFaceSize |
||||
(JNIEnv *, jclass, jlong, jint); |
||||
|
||||
/*
|
||||
* Class: org_opencv_samples_fd_DetectionBaseTracker |
||||
* Method: nativeDetect |
||||
* Signature: (JJJ)V |
||||
*/ |
||||
JNIEXPORT void JNICALL Java_org_opencv_samples_fd_DetectionBaseTracker_nativeDetect |
||||
(JNIEnv *, jclass, jlong, jlong, jlong); |
||||
|
||||
#ifdef __cplusplus |
||||
} |
||||
#endif |
||||
#endif |
||||
@ -0,0 +1,61 @@ |
||||
/* DO NOT EDIT THIS FILE - it is machine generated */ |
||||
#include <jni.h> |
||||
/* Header for class org_opencv_samples_fd_DetectionBasedTracker */ |
||||
|
||||
#ifndef _Included_org_opencv_samples_fd_DetectionBasedTracker |
||||
#define _Included_org_opencv_samples_fd_DetectionBasedTracker |
||||
#ifdef __cplusplus |
||||
extern "C" { |
||||
#endif |
||||
/*
|
||||
* Class: org_opencv_samples_fd_DetectionBasedTracker |
||||
* Method: nativeCreateObject |
||||
* Signature: (Ljava/lang/String;F)J |
||||
*/ |
||||
JNIEXPORT jlong JNICALL Java_org_opencv_samples_fd_DetectionBasedTracker_nativeCreateObject |
||||
(JNIEnv *, jclass, jstring, jint); |
||||
|
||||
/*
|
||||
* Class: org_opencv_samples_fd_DetectionBasedTracker |
||||
* Method: nativeDestroyObject |
||||
* Signature: (J)V |
||||
*/ |
||||
JNIEXPORT void JNICALL Java_org_opencv_samples_fd_DetectionBasedTracker_nativeDestroyObject |
||||
(JNIEnv *, jclass, jlong); |
||||
|
||||
/*
|
||||
* Class: org_opencv_samples_fd_DetectionBasedTracker |
||||
* Method: nativeStart |
||||
* Signature: (J)V |
||||
*/ |
||||
JNIEXPORT void JNICALL Java_org_opencv_samples_fd_DetectionBasedTracker_nativeStart |
||||
(JNIEnv *, jclass, jlong); |
||||
|
||||
/*
|
||||
* Class: org_opencv_samples_fd_DetectionBasedTracker |
||||
* Method: nativeStop |
||||
* Signature: (J)V |
||||
*/ |
||||
JNIEXPORT void JNICALL Java_org_opencv_samples_fd_DetectionBasedTracker_nativeStop |
||||
(JNIEnv *, jclass, jlong); |
||||
|
||||
/*
|
||||
* Class: org_opencv_samples_fd_DetectionBasedTracker |
||||
* Method: nativeSetFaceSize |
||||
* Signature: (JI)V |
||||
*/ |
||||
JNIEXPORT void JNICALL Java_org_opencv_samples_fd_DetectionBasedTracker_nativeSetFaceSize |
||||
(JNIEnv *, jclass, jlong, jint); |
||||
|
||||
/*
|
||||
* Class: org_opencv_samples_fd_DetectionBasedTracker |
||||
* Method: nativeDetect |
||||
* Signature: (JJJ)V |
||||
*/ |
||||
JNIEXPORT void JNICALL Java_org_opencv_samples_fd_DetectionBasedTracker_nativeDetect |
||||
(JNIEnv *, jclass, jlong, jlong, jlong); |
||||
|
||||
#ifdef __cplusplus |
||||
} |
||||
#endif |
||||
#endif |
||||
@ -1,52 +0,0 @@ |
||||
package org.opencv.samples.fd; |
||||
|
||||
import org.opencv.core.Mat; |
||||
import org.opencv.core.MatOfRect; |
||||
|
||||
public class DetectionBaseTracker |
||||
{ |
||||
public DetectionBaseTracker(String filename, int faceSize) |
||||
{ |
||||
mNativeObj = nativeCreateObject(filename, faceSize); |
||||
} |
||||
|
||||
public void start() |
||||
{ |
||||
nativeStart(mNativeObj); |
||||
} |
||||
|
||||
public void stop() |
||||
{ |
||||
nativeStop(mNativeObj); |
||||
} |
||||
|
||||
public void setMinFaceSize(int faceSize) |
||||
{ |
||||
nativeSetFaceSize(mNativeObj, faceSize); |
||||
} |
||||
|
||||
public void detect(Mat imageGray, MatOfRect faces) |
||||
{ |
||||
nativeDetect(mNativeObj, imageGray.getNativeObjAddr(), faces.getNativeObjAddr()); |
||||
} |
||||
|
||||
public void release() |
||||
{ |
||||
nativeDestroyObject(mNativeObj); |
||||
mNativeObj = 0; |
||||
} |
||||
|
||||
protected long mNativeObj = 0; |
||||
|
||||
protected static native long nativeCreateObject(String filename, int faceSize); |
||||
protected static native void nativeDestroyObject(long thiz); |
||||
protected static native void nativeStart(long thiz); |
||||
protected static native void nativeStop(long thiz); |
||||
protected static native void nativeSetFaceSize(long thiz, int faceSize); |
||||
protected static native void nativeDetect(long thiz, long inputImage, long resultMat); |
||||
|
||||
static |
||||
{ |
||||
System.loadLibrary("detection_base_tacker"); |
||||
} |
||||
} |
||||
@ -0,0 +1,52 @@ |
||||
package org.opencv.samples.fd; |
||||
|
||||
import org.opencv.core.Mat; |
||||
import org.opencv.core.MatOfRect; |
||||
|
||||
public class DetectionBasedTracker |
||||
{ |
||||
public DetectionBasedTracker(String cascadeName, int minFaceSize) |
||||
{ |
||||
mNativeObj = nativeCreateObject(cascadeName, minFaceSize); |
||||
} |
||||
|
||||
public void start() |
||||
{ |
||||
nativeStart(mNativeObj); |
||||
} |
||||
|
||||
public void stop() |
||||
{ |
||||
nativeStop(mNativeObj); |
||||
} |
||||
|
||||
public void setMinFaceSize(int size) |
||||
{ |
||||
nativeSetFaceSize(mNativeObj, size); |
||||
} |
||||
|
||||
public void detect(Mat imageGray, MatOfRect faces) |
||||
{ |
||||
nativeDetect(mNativeObj, imageGray.getNativeObjAddr(), faces.getNativeObjAddr()); |
||||
} |
||||
|
||||
public void release() |
||||
{ |
||||
nativeDestroyObject(mNativeObj); |
||||
mNativeObj = 0; |
||||
} |
||||
|
||||
private long mNativeObj = 0; |
||||
|
||||
private static native long nativeCreateObject(String cascadeName, int minFaceSize); |
||||
private static native void nativeDestroyObject(long thiz); |
||||
private static native void nativeStart(long thiz); |
||||
private static native void nativeStop(long thiz); |
||||
private static native void nativeSetFaceSize(long thiz, int size); |
||||
private static native void nativeDetect(long thiz, long inputImage, long faces); |
||||
|
||||
static |
||||
{ |
||||
System.loadLibrary("detection_based_tacker"); |
||||
} |
||||
} |
||||
Loading…
Reference in new issue