c++ MTCNN

https://github.com/szad670401/Fast-MTCNN

70ms一张，可能用的cpu

  

   

    

     

    
    

     

      //Created by Jack Yu
     
    
   

    

     

    
    

     

      #include "stdafx.h"
     
    
   

    

     

    
    

     

      #include <fstream>
     
    
   

    

     

    
    

     

      #include <iostream>
     
    
   

    

     

    
    

     

      #include <opencv2/opencv.hpp>
     
    
   

    

     

    
    

     

      #include <opencv2/dnn.hpp>
     
    
   

    

     

    
    

     

      using namespace std;
     
    
   

    

     

    
    

     

      using namespace cv;
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      const float pnet_stride = 2;
     
    
   

    

     

    
    

     

      const float pnet_cell_size = 12;
     
    
   

    

     

    
    

     

      const int pnet_max_detect_num = 5000;
     
    
   

    

     

    
    

     

      //mean & std
     
    
   

    

     

    
    

     

      const float mean_val = 127.5f;
     
    
   

    

     

    
    

     

      const float std_val = 0.0078125f;
     
    
   

    

     

    
    

     

      //minibatch size
     
    
   

    

     

    
    

     

      const int step_size = 128;
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      typedef struct FaceBox {
     
    
   

    

     

    
    

     
	float xmin;
     
    
   

    

     

    
    

     
	float ymin;
     
    
   

    

     

    
    

     
	float xmax;
     
    
   

    

     

    
    

     
	float ymax;
     
    
   

    

     

    
    

     
	float score;
     
    
   

    

     

    
    

     

      } FaceBox;
     
    
   

    

     

    
    

     

      typedef struct FaceInfo {
     
    
   

    

     

    
    

     
	float bbox_reg[4];
     
    
   

    

     

    
    

     
	float landmark_reg[10];
     
    
   

    

     

    
    

     
	float landmark[10];
     
    
   

    

     

    
    

     

      	FaceBox bbox;
     
    
   

    

     

    
    

     

      } FaceInfo;
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      class MTCNN {
     
    
   

    

     

    
    

     

      public:
     
    
   

    

     

    
    

     

      	MTCNN(const string& proto_model_dir);
     
    
   

    

     

    
    

     
	vector<FaceInfo> Detect_mtcnn(const cv::Mat& img,
     
    
  
 

文章来源: blog.csdn.net，作者：网奇，版权归原作者所有，如需转载，请联系作者。

原文链接：blog.csdn.net/jacke121/article/details/100901217