hogPeopleDetection.java

// ********************************************************

// Example : displaying live from attached camera
// and (try to!) detect people in it using HOG

// Author : Toby Breckon, toby.breckon@durham.ac.uk

// Copyright (c) 2015 Durham University
// License : LGPL - http://www.gnu.org/licenses/lgpl.html

// ********************************************************

// import required OpenCV components

import java.util.List;

import org.opencv.core.Core;
import org.opencv.core.Mat;
import org.opencv.core.MatOfDouble;
import org.opencv.core.MatOfRect;
import org.opencv.core.Point;
import org.opencv.core.Rect;
import org.opencv.core.Scalar;
import org.opencv.core.Size;
import org.opencv.highgui.VideoCapture;
import org.opencv.imgproc.Imgproc;
import org.opencv.objdetect.HOGDescriptor;

// ********************************************************

public class hogPeopleDetection {

    public static void main(String[] args) throws InterruptedException {

        // load the Core OpenCV library by name

        System.loadLibrary(Core.NATIVE_LIBRARY_NAME);

        // create video capture device object

        VideoCapture cap = new VideoCapture();

        // try to use the hardware device if present
        
        int CAM_TO_USE = 0;
        
        // create a new image object
        
        Mat matFrame = new Mat();
        
        // try to open first capture device (0)
        
        try {
        	cap.open(CAM_TO_USE);
        } catch (Exception e1) {
            System.out.println("No webcam attached");
        
            // otherwise try opening a video file 
            
            try{
            	cap.open("files/video.mp4");
            } catch (Exception e2) {
            	System.out.println("No video file found");
            }
        }
        
        // if the a video capture source is now open
        
        if (cap.isOpened())
        {

            // create a new window object
            
            Imshow ims = new Imshow("HOG People Detection");
        	
        	boolean keepProcessing = true;

        	// create new HoG detection object
        	
        	HOGDescriptor HOG = new HOGDescriptor();
        	HOG.setSVMDetector(HOGDescriptor.getDefaultPeopleDetector());
        	
        	// create 
        	
        	MatOfRect foundLocations = new MatOfRect();
        	MatOfDouble foundWeights = new MatOfDouble();
        	
        	while (keepProcessing)
        	{	
        		// grab the next frame from video source

        		cap.grab();

            	// decode and return the grabbed video frame

            	cap.retrieve(matFrame);

            	// if the frame is valid (not end of video for example)
            	
            	if (!(matFrame.empty()))
            	{
            		// perform detection

					HOG.detectMultiScale(matFrame, foundLocations, foundWeights,
                            0,
                            new Size(8,8),
                            new Size(32,32),
                            1.05,
                            8,
                            false);
            		
					List<Rect> rectangles = foundLocations.toList();
					
					for (int i = 0; i < rectangles.size(); i++) {
			        	
						Core.rectangle(matFrame,
					             new Point(rectangles.get(i).x,rectangles.get(i).y),
					             new Point(rectangles.get(i).x + rectangles.get(i).width,
					            		   rectangles.get(i).y + rectangles.get(i).height),
					             new Scalar (255,0,0), 2, 1, 0);
			        }
					
            		// display image with a delay of 40ms (i.e. 1000 ms / 25 = 25 fps)

            		ims.showImage(matFrame);
            		Thread.sleep(40);
            	} else { 
            		keepProcessing = false;
        		}
        	}

        } else {
            System.out.println("error cannot open any capture source - exiting");
        }

        // close down the camera correctly

        cap.release();
        
    }
}

// ********************************************************