samples/cpp/peopledetect.cpp

   1 #include <iostream>
   2 #include <stdexcept>
   3 #include <opencv2/objdetect.hpp>
   4 #include <opencv2/highgui.hpp>
   5 #include <opencv2/imgproc.hpp>
   6 #include <opencv2/imgcodecs.hpp>
   7 #include <opencv2/video.hpp>
   8 #include <opencv2/videoio.hpp>
   9
  10 using namespace cv;
  11 using namespace std;
  12
  13
  14 const char* keys =
  15 {
  16     "{ help h      |                     | print help message }"
  17     "{ image i     |                     | specify input image}"
  18     "{ camera c    |                     | enable camera capturing }"
  19     "{ video v     | ../data/vtest.avi   | use video as input }"
  20     "{ directory d |                     | images directory}"
  21 };
  22
  23 static void detectAndDraw(const HOGDescriptor &hog, Mat &img)
  24 {
  25     vector<Rect> found, found_filtered;
  26     double t = (double) getTickCount();
  27     // Run the detector with default parameters. to get a higher hit-rate
  28     // (and more false alarms, respectively), decrease the hitThreshold and
  29     // groupThreshold (set groupThreshold to 0 to turn off the grouping completely).
  30     hog.detectMultiScale(img, found, 0, Size(8,8), Size(32,32), 1.05, 2);
  31     t = (double) getTickCount() - t;
  32     cout << "detection time = " << (t*1000./cv::getTickFrequency()) << " ms" << endl;
  33
  34     for(size_t i = 0; i < found.size(); i++ )
  35     {
  36         Rect r = found[i];
  37
  38         size_t j;
  39         // Do not add small detections inside a bigger detection.
  40         for ( j = 0; j < found.size(); j++ )
  41             if ( j != i && (r & found[j]) == r )
  42                 break;
  43
  44         if ( j == found.size() )
  45             found_filtered.push_back(r);
  46     }
  47
  48     for (size_t i = 0; i < found_filtered.size(); i++)
  49     {
  50         Rect r = found_filtered[i];
  51
  52         // The HOG detector returns slightly larger rectangles than the real objects,
  53         // so we slightly shrink the rectangles to get a nicer output.
  54         r.x += cvRound(r.width*0.1);
  55         r.width = cvRound(r.width*0.8);
  56         r.y += cvRound(r.height*0.07);
  57         r.height = cvRound(r.height*0.8);
  58         rectangle(img, r.tl(), r.br(), cv::Scalar(0,255,0), 3);
  59     }
  60 }
  61
  62 int main(int argc, char** argv)
  63 {
  64     CommandLineParser parser(argc, argv, keys);
  65
  66     if (parser.has("help"))
  67     {
  68         cout << "\nThis program demonstrates the use of the HoG descriptor using\n"
  69             " HOGDescriptor::hog.setSVMDetector(HOGDescriptor::getDefaultPeopleDetector());\n";
  70         parser.printMessage();
  71         cout << "During execution:\n\tHit q or ESC key to quit.\n"
  72             "\tUsing OpenCV version " << CV_VERSION << "\n"
  73             "Note: camera device number must be different from -1.\n" << endl;
  74         return 0;
  75     }
  76
  77     HOGDescriptor hog;
  78     hog.setSVMDetector(HOGDescriptor::getDefaultPeopleDetector());
  79     namedWindow("people detector", 1);
  80
  81     string pattern_glob = "";
  82     string video_filename = "../data/vtest.avi";
  83     int camera_id = -1;
  84     if (parser.has("directory"))
  85     {
  86         pattern_glob = parser.get<string>("directory");
  87     }
  88     else if (parser.has("image"))
  89     {
  90         pattern_glob = parser.get<string>("image");
  91     }
  92     else if (parser.has("camera"))
  93     {
  94         camera_id = parser.get<int>("camera");
  95     }
  96     else if (parser.has("video"))
  97     {
  98         video_filename = parser.get<string>("video");
  99     }
 100
 101     if (!pattern_glob.empty() || camera_id != -1 || !video_filename.empty())
 102     {
 103         //Read from input image files
 104         vector<String> filenames;
 105         //Read from video file
 106         VideoCapture vc;
 107         Mat frame;
 108
 109         if (!pattern_glob.empty())
 110         {
 111             String folder(pattern_glob);
 112             glob(folder, filenames);
 113         }
 114         else if (camera_id != -1)
 115         {
 116             vc.open(camera_id);
 117             if (!vc.isOpened())
 118             {
 119                 stringstream msg;
 120                 msg << "can't open camera: " << camera_id;
 121                 throw runtime_error(msg.str());
 122             }
 123         }
 124         else
 125         {
 126             vc.open(video_filename.c_str());
 127             if (!vc.isOpened())
 128                 throw runtime_error(string("can't open video file: " + video_filename));
 129         }
 130
 131         vector<String>::const_iterator it_image = filenames.begin();
 132
 133         for (;;)
 134         {
 135             if (!pattern_glob.empty())
 136             {
 137                 bool read_image_ok = false;
 138                 for (; it_image != filenames.end(); ++it_image)
 139                 {
 140                     cout << "\nRead: " << *it_image << endl;
 141                     // Read current image
 142                     frame = imread(*it_image);
 143
 144                     if (!frame.empty())
 145                     {
 146                         ++it_image;
 147                         read_image_ok = true;
 148                         break;
 149                     }
 150                 }
 151
 152                 //No more valid images
 153                 if (!read_image_ok)
 154                 {
 155                     //Release the image in order to exit the while loop
 156                     frame.release();
 157                 }
 158             }
 159             else
 160             {
 161                 vc >> frame;
 162             }
 163
 164             if (frame.empty())
 165                 break;
 166
 167             detectAndDraw(hog, frame);
 168
 169             imshow("people detector", frame);
 170             int c = waitKey( vc.isOpened() ? 30 : 0 ) & 255;
 171             if ( c == 'q' || c == 'Q' || c == 27)
 172                 break;
 173         }
 174     }
 175
 176     return 0;
 177 }