picture.cpp

#include "PCN.h"
#include "mropencv.h"
int main0()
{
    PCN detector("model/PCN.caffemodel",
                 "model/PCN-1.prototxt", "model/PCN-2.prototxt", "model/PCN-3.prototxt");
    detector.SetMinFaceSize(45);
    detector.SetScoreThresh(0.37, 0.43, 0.95);
    detector.SetImagePyramidScaleFactor(1.414);
    detector.SetVideoSmooth(false);

    for (int i = 1; i <= 26; i++)
    {
        cv::Mat img =
            cv::imread("../imgs/" + std::to_string(i) + ".jpg");
        cv::TickMeter tm;
        tm.reset();
        tm.start();
        std::vector<Window> faces = detector.DetectFace(img);
        tm.stop();
        std::cout << "Image: " << i << std::endl;
        std::cout << "Time Cost: "<<
                  tm.getTimeMilli() << " ms" << std::endl;
        for (int i = 0; i < faces.size(); i++)
        {
            DrawFace(img, faces[i]);
        }
        cv::imshow("PCN", img);
        cv::waitKey();
    }

    cv::destroyAllWindows();

    return 0;
}

int main()
{
	PCN detector("model/PCN.caffemodel",
		"model/PCN-1.prototxt", "model/PCN-2.prototxt", "model/PCN-3.prototxt");
	detector.SetMinFaceSize(45);
	detector.SetScoreThresh(0.37, 0.43, 0.95);
	detector.SetImagePyramidScaleFactor(1.414);
	detector.SetVideoSmooth(false);
	cv::VideoCapture capture(0);
	cv::Mat img;
	while (true)
	{
		capture >> img;
		if (img.empty())
			break;
		cv::TickMeter tm;
		tm.reset();
		tm.start();
		std::vector<Window> faces = detector.DetectFace(img);
		tm.stop();
		std::cout << "Time Cost: " <<
			tm.getTimeMilli() << " ms" << std::endl;
		for (int i = 0; i < faces.size(); i++)
		{
			DrawFace(img, faces[i]);
		}
		cv::imshow("PCN", img);
		cv::waitKey(1);
	}
	return 0;
}