#include <iostream>
|
#include <vector>
|
#include <stdio.h>
|
#include <cv_face.h>
|
#include "time_helper.h"
|
|
#include <opencv2/opencv.hpp>
|
#define DEFAULT_THRESHOLD (0.5)
|
|
using namespace std;
|
using namespace cv;
|
|
//ÈËÁ³ÑéÖ¤
|
static cv_handle_t handle_verify = NULL;
|
//Êý¾Ý¿â²Ù×÷
|
static cv_handle_t handle_db = NULL;
|
//¾²Ì¬ÈËÁ³Ê¶±ð
|
static cv_handle_t handle_detect = NULL;
|
|
#define _MAX_PATH 260
|
|
//ÌáȡͼƬÌØÕ÷Öµ
|
cv_feature_t *extract_feature(Mat image_color) {
|
|
Mat image_color_color;
|
// cvtColor(image_color, image_color_color, CV_BGR2BGRA); // CV_PIX_FMT_BGRA8888
|
image_color_color = image_color; // CV_PIX_FMT_BGR888
|
|
cv_feature_t *p_feature = NULL;
|
cv_face_t *p_face = NULL;
|
int face_count = 0;
|
cv_result_t st_result = CV_OK;
|
//¼ì²âÈËÁ³
|
st_result = cv_face_detect(handle_detect, image_color_color.data, CV_PIX_FMT_BGR888,
|
image_color_color.cols, image_color_color.rows, image_color_color.step,
|
CV_FACE_UP, &p_face, &face_count);
|
if (face_count >= 1) {
|
//»ñÈ¡ÈËÁ³ÌØÕ÷
|
st_result = cv_verify_get_feature(handle_verify,
|
(unsigned char *)image_color_color.data, CV_PIX_FMT_BGR888,
|
image_color_color.cols, image_color_color.rows, image_color_color.step,
|
p_face, &p_feature, NULL);
|
if (st_result != CV_OK) {
|
fprintf(stderr, "cv_verify_get_feature failed, error code %d\n", st_result);
|
}
|
} else {
|
fprintf(stderr, "can't find face in ");
|
}
|
// ÊÍ·ÅÄÚ´æ
|
cv_face_release_detector_result(p_face, face_count);
|
return p_feature;
|
}
|
|
//Êý¾Ý¿âÌí¼ÓÊý¾Ý£¬²¢·µ»Ø¼Ç¼µÃid
|
int db_add(cv_feature_t *p_feature) {
|
/*cv_feature_t *p_feature = extract_feature(image_color);
|
if (!p_feature) {
|
return -1;
|
}*/
|
int idx;
|
cv_result_t cv_result = cv_verify_add_face(handle_db, p_feature, &idx);
|
if (cv_result != CV_OK) {
|
fprintf(stderr, "cv_verify_add_face failed, error code %d\n", cv_result);
|
}
|
cv_verify_release_feature(p_feature);
|
return idx;
|
}
|
|
//¸ù¾Ýidɾ³ý¼Ç¼
|
bool db_del(int idx) {
|
if (idx < 0) {
|
fprintf(stderr, "invalid idx!\n");
|
return false;
|
}
|
cv_result_t cv_result = CV_OK;
|
cv_result = cv_verify_delete_face(handle_db, idx);
|
if (cv_result != CV_OK) {
|
fprintf(stderr, "cv_verify_delete_face failed, error code %d\n", cv_result);
|
} else {
|
fprintf(stderr, "delete succeed\n");
|
}
|
}
|
|
//Êý¾Ý¿â±£´æ
|
bool db_save(char *db_path) {
|
cv_result_t cv_result = CV_OK;
|
cv_result = cv_verify_save_db(handle_db, db_path);
|
if (cv_result != CV_OK) {
|
fprintf(stderr, "cv_verify_save_db failed, error code %d\n", cv_result);
|
return false;
|
} else {
|
fprintf(stderr, "save done!\n");
|
}
|
|
return true;
|
}
|
|
//Êý¾Ý¿â¼ÓÔØ
|
bool db_load(char *db_path) {
|
cv_result_t cv_result = CV_OK;
|
cv_result = cv_verify_load_db(handle_db, db_path);
|
|
if (cv_result != CV_OK) {
|
fprintf(stderr, "cv_verify_load_db failed, error code %d\n", cv_result);
|
return false;
|
} else {
|
fprintf(stderr, "load done!\n");
|
}
|
|
return true;
|
}
|
|
//ËÑË÷Êý¾Ý¿â
|
bool search_db(Mat image_color,char *db_path) {
|
|
cv_feature_t *p_feature = extract_feature(image_color);
|
if (p_feature == NULL) {
|
fprintf(stderr, "extract failed !\n");
|
return false;
|
}
|
db_load(db_path);
|
//²éѯǰ10Ìõ
|
int top_k = 10;
|
int *top_idxs = new int[top_k];
|
float *top_scores = new float[top_k];
|
unsigned int result_length = 0;
|
cv_result_t cv_result = cv_verify_search_face(handle_verify, handle_db,
|
p_feature, top_k,
|
top_idxs, top_scores, &result_length);
|
if (cv_result == CV_OK) {
|
//Êä³ö¼ì²é½á¹û
|
for (unsigned int t = 0; t < result_length; t++) {
|
// const cv_feature_t item = result[t].item;
|
fprintf(stderr, "%d\t", top_idxs[t]);
|
fprintf(stderr, "%0.2f\n", top_scores[t]);
|
}
|
} else {
|
fprintf(stderr, "cv_verify_search_face failed, error code %d\n", cv_result);
|
}
|
if (top_idxs) {
|
delete[]top_idxs;
|
}
|
if (top_scores) {
|
delete[]top_scores;
|
}
|
cv_verify_release_feature(p_feature);
|
return true;
|
}
|
|
//ÅúÁ¿µ¼ÈëͼƬ½øÐзÖÎö
|
bool db_gen(char *image_list, char *db_path) {
|
bool bresult = true;
|
//¶ÁȡͼƬÁбí
|
FILE *fp_path = fopen(image_list, "r");
|
if(!fp_path) {
|
fprintf(stderr, "failed to load %s\n", image_list);
|
return false;
|
}
|
|
//ÈËÁ³ÌØÕ÷ÐÅÏ¢ Êý¾Ý½á¹¹
|
std::vector<cv_feature_t *> list_feature;
|
list_feature.clear();
|
|
for (;;) {
|
char image_path[1024];
|
int num = fscanf(fp_path, "%s", image_path);
|
if (num != 1) {
|
bresult = false;
|
break;
|
}
|
fprintf(stderr, "extracting %s\n", image_path);
|
|
|
// get the face feature
|
Mat image_color = imread(image_path);
|
if (!image_color.data) {
|
return NULL;
|
}
|
|
cv_feature_t *p_feature = extract_feature(image_color);
|
if (!p_feature) {
|
fprintf(stderr, "failed to extract image: %s\n", image_path);
|
continue;
|
}
|
list_feature.push_back(p_feature);
|
}
|
fclose(fp_path);
|
|
|
cv_verify_destroy_db(handle_db);
|
cv_result_t cv_result = CV_OK;
|
cv_verify_create_db(&handle_db);
|
//´´½¨ÐµÄÊý¾Ý¿â£¬Ó¦¸Ã¸ÄΪÅжϣ¬Èç¹ûÊý¾Ý¿â´æÔÚÔò½øÐвåÈ룬·ñÔòн¨¡£
|
if(db_load(db_path)){
|
for (int x = 0; x < list_feature.size(); x++) {
|
db_add(list_feature[x]);
|
}
|
}else{
|
cv_result = cv_verify_build_db(handle_db, &list_feature[0], list_feature.size());
|
if (cv_result != CV_OK) {
|
fprintf(stderr, "cv_verify_build_db failed, error code %d\n", cv_result);
|
bresult = false;
|
}
|
for (int i = 0; i < list_feature.size(); i++) {
|
cv_verify_release_feature(list_feature[i]);
|
}
|
}
|
cv_verify_save_db(handle_db, db_path);
|
|
|
return bresult;
|
}
|
|
int main(int argc, char *argv[]) {
|
|
//Îļþ·¾¶
|
char *db_path = "./out1.db";
|
char *image_path = "../../test_image/face_08.jpg";
|
char *image_list = "../../test_image/imglist";
|
|
//´´½¨¾²Ì¬ÈËÁ³Ê¶±ð¾ä±ú£¬Í¬Ê±´´½¨×´Ì¬±êÁ¿
|
cv_result_t cv_result = cv_face_create_detector(&handle_detect, NULL, CV_DETECT_ENABLE_ALIGN_21);
|
if (cv_result != CV_OK){
|
fprintf(stderr, "create detect handle failed, error code %d\n", cv_result);
|
}
|
//´´½¨ÈËÁ³ÑéÖ¤¾ä±ú²¢³õʼ»¯
|
cv_result = cv_verify_create_handle(&handle_verify, "../../../models/verify.model");
|
if (cv_result != CV_OK){
|
fprintf(stderr, "create verify handle failed, error code %d\n", cv_result);
|
}
|
// ´´½¨ÈËÁ³Êý¾Ý¿â¾ä±ú
|
cv_result = cv_verify_create_db(&handle_db);
|
if (cv_result != CV_OK){
|
fprintf(stderr, "create db handle failed, error code %d\n", cv_result);
|
}
|
|
//ÅúÁ¿²åÈëÊý¾Ý¿â
|
db_gen(image_list,db_path);
|
|
//´ýËÑË÷µÄͼÏñ
|
Mat image_color = imread(image_path);
|
if (!image_color.data) {
|
return NULL;
|
}
|
//ËÑË÷Êý¾Ý¿â
|
if(search_db(image_color,db_path)){
|
cout<<"search db"<<endl;
|
}
|
|
//»ØÊÕÄÚ´æ
|
cv_face_destroy_detector(handle_detect);
|
cv_verify_destroy_db(handle_db);
|
cv_verify_destroy_handle(handle_verify);
|
return 0;
|
}
|
