ObjectDetectionDemo.java 3.5 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798
  1. import java.util.List;
  2. import org.opencv.core.Core;
  3. import org.opencv.core.Mat;
  4. import org.opencv.core.MatOfRect;
  5. import org.opencv.core.Point;
  6. import org.opencv.core.Rect;
  7. import org.opencv.core.Scalar;
  8. import org.opencv.core.Size;
  9. import org.opencv.highgui.HighGui;
  10. import org.opencv.imgproc.Imgproc;
  11. import org.opencv.objdetect.CascadeClassifier;
  12. import org.opencv.videoio.VideoCapture;
  13. class ObjectDetection {
  14. public void detectAndDisplay(Mat frame, CascadeClassifier faceCascade, CascadeClassifier eyesCascade) {
  15. Mat frameGray = new Mat();
  16. Imgproc.cvtColor(frame, frameGray, Imgproc.COLOR_BGR2GRAY);
  17. Imgproc.equalizeHist(frameGray, frameGray);
  18. // -- Detect faces
  19. MatOfRect faces = new MatOfRect();
  20. faceCascade.detectMultiScale(frameGray, faces);
  21. List<Rect> listOfFaces = faces.toList();
  22. for (Rect face : listOfFaces) {
  23. Point center = new Point(face.x + face.width / 2, face.y + face.height / 2);
  24. Imgproc.ellipse(frame, center, new Size(face.width / 2, face.height / 2), 0, 0, 360,
  25. new Scalar(255, 0, 255));
  26. Mat faceROI = frameGray.submat(face);
  27. // -- In each face, detect eyes
  28. MatOfRect eyes = new MatOfRect();
  29. eyesCascade.detectMultiScale(faceROI, eyes);
  30. List<Rect> listOfEyes = eyes.toList();
  31. for (Rect eye : listOfEyes) {
  32. Point eyeCenter = new Point(face.x + eye.x + eye.width / 2, face.y + eye.y + eye.height / 2);
  33. int radius = (int) Math.round((eye.width + eye.height) * 0.25);
  34. Imgproc.circle(frame, eyeCenter, radius, new Scalar(255, 0, 0), 4);
  35. }
  36. }
  37. //-- Show what you got
  38. HighGui.imshow("Capture - Face detection", frame );
  39. }
  40. public void run(String[] args) {
  41. String filenameFaceCascade = args.length > 2 ? args[0] : "../../data/haarcascades/haarcascade_frontalface_alt.xml";
  42. String filenameEyesCascade = args.length > 2 ? args[1] : "../../data/haarcascades/haarcascade_eye_tree_eyeglasses.xml";
  43. int cameraDevice = args.length > 2 ? Integer.parseInt(args[2]) : 0;
  44. CascadeClassifier faceCascade = new CascadeClassifier();
  45. CascadeClassifier eyesCascade = new CascadeClassifier();
  46. if (!faceCascade.load(filenameFaceCascade)) {
  47. System.err.println("--(!)Error loading face cascade: " + filenameFaceCascade);
  48. System.exit(0);
  49. }
  50. if (!eyesCascade.load(filenameEyesCascade)) {
  51. System.err.println("--(!)Error loading eyes cascade: " + filenameEyesCascade);
  52. System.exit(0);
  53. }
  54. VideoCapture capture = new VideoCapture(cameraDevice);
  55. if (!capture.isOpened()) {
  56. System.err.println("--(!)Error opening video capture");
  57. System.exit(0);
  58. }
  59. Mat frame = new Mat();
  60. while (capture.read(frame)) {
  61. if (frame.empty()) {
  62. System.err.println("--(!) No captured frame -- Break!");
  63. break;
  64. }
  65. //-- 3. Apply the classifier to the frame
  66. detectAndDisplay(frame, faceCascade, eyesCascade);
  67. if (HighGui.waitKey(10) == 27) {
  68. break;// escape
  69. }
  70. }
  71. System.exit(0);
  72. }
  73. }
  74. public class ObjectDetectionDemo {
  75. public static void main(String[] args) {
  76. // Load the native OpenCV library
  77. System.loadLibrary(Core.NATIVE_LIBRARY_NAME);
  78. new ObjectDetection().run(args);
  79. }
  80. }