main.cpp 1.09 KB
#include "facecomparison.h"
int main(){
    
    Mat image1 = cv::imread("/data/face_recognize/pipeline_test/35326a760af111ec9f2300163e772630/310cardImageContent163047910320427.jpg");
    Mat image2 = cv::imread("/data/face_recognize/pipeline_test/35326a760af111ec9f2300163e772630/310faceImageContent163047910353714.jpg");
    string face_det_model = "/home/situ/qfs/sdk_project/mnn_projects/face_recognize_mnn/model/det_face_retina_mnn_1.0.0_v0.1.1.mnn";
    string face_landm_model = "/home/situ/qfs/sdk_project/mnn_projects/face_recognize_mnn/model/det_landmarks_106_v0.0.1.mnn";
    string face_rec_model = "/home/situ/qfs/sdk_project/mnn_projects/face_recognize_mnn/model/cls_face_mnn_1.0.0_v0.1.1.mnn";

    InitModelError init_model_code;
    OutputInfo output_info;
    FaceComparison face_rec = FaceComparison();
    bool init_ = face_rec.init_model(face_det_model,face_landm_model,face_rec_model,init_model_code);
    if(init_){
        int result = face_rec.inference(image1,image2,output_info);
        cout<<output_info.error_code<<" "<<output_info.result<<endl;
    }else{
        return -1;
    }

    
}