samples/cpp/peopledetect.cpp

// 此檔案是 OpenCV 專案的一部分。
// 它受許可條款的約束，許可條款位於頂層目錄中的 LICENSE 檔案中
// 此發行的版本中，網址為 https://opencv.tw/license.html
 
#include <opencv2/objdetect.hpp>
#include <opencv2/highgui.hpp>
#include <opencv2/imgproc.hpp>
#include <opencv2/videoio.hpp>
#include <iostream>
#include <iomanip>
 
using namespace cv;
using namespace std;
 
class Detector
{
    enum Mode { Default, Daimler } m;
    HOGDescriptor hog, hog_d;
public:
Detector() : m(Default), hog(), hog_d(Size(48, 96), Size(16, 16), Size(8, 8), Size(8, 8), 9)
    {
hog.setSVMDetector(HOGDescriptor::getDefaultPeopleDetector());
hog_d.setSVMDetector(HOGDescriptor::getDaimlerPeopleDetector());
    }
    void toggleMode() { m = (m == Default ? Daimler : Default); }
    string modeName() const { return (m == Default ? "Default" : "Daimler"); }
vector<Rect> detect(InputArray img)
    {
        // 使用預設引數執行檢測器。為了獲得更高的命中率
        // (以及更多的誤報，分別)，降低 hitThreshold 和
        // groupThreshold（將 groupThreshold 設定為 0 以完全關閉分組）。
vector<Rect> found;
        if (m == Default)
hog.detectMultiScale(img, found, 0, Size(8,8), Size(), 1.05, 2, false);
        else if (m == Daimler)
hog_d.detectMultiScale(img, found, 0, Size(8,8), Size(), 1.05, 2, true);
        return found;
    }
    void adjustRect(Rect & r) const
    {
        // HOG 檢測器返回的矩形略大於實際物件，
        // 因此我們稍微縮小矩形以獲得更好的輸出。
r.x += cvRound(r.width*0.1);
r.width = cvRound(r.width*0.8);
r.y += cvRound(r.height*0.07);
r.height = cvRound(r.height*0.8);
    }
};
 
static const string keys = "{ help h | | 列印幫助資訊 }"
                           "{ camera c | 0 | 從攝像頭捕獲影片（裝置索引從 0 開始）}"
                           "{ video v | | 使用影片作為輸入 }";
 
int main(int argc, char** argv)
{
    CommandLineParser parser(argc, argv, keys);
parser.about("此示例演示了 HoG 描述符的使用。");
    if (parser.has("help"))
    {
parser.printMessage();
        return 0;
    }
    int camera = parser.get<int>("camera");
    string file = parser.get<string>("video");
    if (!parser.check())
    {
parser.printErrors();
        return 1;
    }
 
    VideoCapture cap;
    if (file.empty())
cap.open(camera);
    else
    {
file = samples::findFileOrKeep(file);
cap.open(file);
    }
    if (!cap.isOpened())
    {
cout << "無法開啟影片流：'" << (file.empty() ? "<camera>" : file) << "'" << endl;
        return 2;
    }
 
cout << "按 'q' 或 <ESC> 退出。" << endl;
cout << "按 <空格鍵> 在 Default 和 Daimler 檢測器之間切換" << endl;
Detector detector;
    Mat frame;
    for (;;)
    {
cap >> frame;
        if (frame.empty())
        {
cout << "讀取完成：空幀" << endl;
            break;
        }
        int64 t = getTickCount();
vector<Rect> found = detector.detect(frame);
t = getTickCount() - t;
 
        // 顯示視窗
        {
ostringstream buf;
buf << "模式: " << detector.modeName() << " ||| "
<< "FPS: " << fixed << setprecision(1) << (getTickFrequency() / (double)t);
            putText(frame, buf.str(), Point(10, 30), FONT_HERSHEY_PLAIN, 2.0, Scalar(0, 0, 255), 2, LINE_AA);
        }
        for (vector<Rect>::iterator i = found.begin(); i != found.end(); ++i)
        {
            Rect &r = *i;
detector.adjustRect(r);
            rectangle(frame, r.tl(), r.br(), cv::Scalar(0, 255, 0), 2);
        }
        imshow("人體檢測器", frame);
 
        // 與使用者互動
        const char key = (char)waitKey(1);
        if (key == 27 || key == 'q') // ESC
        {
cout << "已請求退出" << endl;
            break;
        }
        else if (key == ' ')
        {
detector.toggleMode();
        }
    }
    return 0;
}