OpticalFlowDenseDemo.java 2.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172
  1. import java.util.ArrayList;
  2. import org.opencv.core.*;
  3. import org.opencv.highgui.HighGui;
  4. import org.opencv.imgproc.Imgproc;
  5. import org.opencv.video.Video;
  6. import org.opencv.videoio.VideoCapture;
  7. class OptFlowDense {
  8. public void run(String[] args) {
  9. String filename = args[0];
  10. VideoCapture capture = new VideoCapture(filename);
  11. if (!capture.isOpened()) {
  12. //error in opening the video input
  13. System.out.println("Unable to open file!");
  14. System.exit(-1);
  15. }
  16. Mat frame1 = new Mat() , prvs = new Mat();
  17. capture.read(frame1);
  18. Imgproc.cvtColor(frame1, prvs, Imgproc.COLOR_BGR2GRAY);
  19. while (true) {
  20. Mat frame2 = new Mat(), next = new Mat();
  21. capture.read(frame2);
  22. if (frame2.empty()) {
  23. break;
  24. }
  25. Imgproc.cvtColor(frame2, next, Imgproc.COLOR_BGR2GRAY);
  26. Mat flow = new Mat(prvs.size(), CvType.CV_32FC2);
  27. Video.calcOpticalFlowFarneback(prvs, next, flow,0.5,3,15,3,5,1.2,0);
  28. // visualization
  29. ArrayList<Mat> flow_parts = new ArrayList<>(2);
  30. Core.split(flow, flow_parts);
  31. Mat magnitude = new Mat(), angle = new Mat(), magn_norm = new Mat();
  32. Core.cartToPolar(flow_parts.get(0), flow_parts.get(1), magnitude, angle,true);
  33. Core.normalize(magnitude, magn_norm,0.0,1.0, Core.NORM_MINMAX);
  34. float factor = (float) ((1.0/360.0)*(180.0/255.0));
  35. Mat new_angle = new Mat();
  36. Core.multiply(angle, new Scalar(factor), new_angle);
  37. //build hsv image
  38. ArrayList<Mat> _hsv = new ArrayList<>() ;
  39. Mat hsv = new Mat(), hsv8 = new Mat(), bgr = new Mat();
  40. _hsv.add(new_angle);
  41. _hsv.add(Mat.ones(angle.size(), CvType.CV_32F));
  42. _hsv.add(magn_norm);
  43. Core.merge(_hsv, hsv);
  44. hsv.convertTo(hsv8, CvType.CV_8U, 255.0);
  45. Imgproc.cvtColor(hsv8, bgr, Imgproc.COLOR_HSV2BGR);
  46. HighGui.imshow("frame2", bgr);
  47. int keyboard = HighGui.waitKey(30);
  48. if (keyboard == 'q' || keyboard == 27) {
  49. break;
  50. }
  51. prvs = next;
  52. }
  53. System.exit(0);
  54. }
  55. }
  56. public class OpticalFlowDenseDemo {
  57. public static void main(String[] args) {
  58. System.loadLibrary(Core.NATIVE_LIBRARY_NAME);
  59. new OptFlowDense().run(args);
  60. }
  61. }