首页 > 代码库 > 【计算机视觉】OpenCV人脸识别facerec源码分析1——FaceRecognizer概述
【计算机视觉】OpenCV人脸识别facerec源码分析1——FaceRecognizer概述
FaceRecognizer类
OpenCV中所有人脸识别的模型都是继承自FaceRecognizer这个基类,这个类提供了人脸识别算法的统一接口。
class FaceRecognizer : public Algorithm
{
public:
//! virtual destructor
virtual ~FaceRecognizer() {}
// Trains a FaceRecognizer.
virtual void train(InputArray src, InputArray labels) = 0;
// Updates a FaceRecognizer.
virtual void update(InputArrayOfArrays src, InputArray labels);
// Gets a prediction from a FaceRecognizer.
virtual int predict(InputArray src) const = 0;
// Predicts the label and confidence for a given sample.
virtual void predict(InputArray src, int &label, double &confidence) const = 0;
// Serializes this object to a given filename.
virtual void save(const string& filename) const;
// Deserializes this object from a given filename.
virtual void load(const string& filename);
// Serializes this object to a given cv::FileStorage.
virtual void save(FileStorage& fs) const = 0;
// Deserializes this object from a given cv::FileStorage.
virtual void load(const FileStorage& fs) = 0;
};
Algorithm类
Algorithm所有的派生类都提供了一下功能:
- 虚拟构造器。每个继承Algorithm的算法在程序开始时注册,你可以得到已注册算法的列表,并且根据指定算法名字来创建一个实例。如果你想添加自己的算法,比较好的方法是在你算法名上添加一个前缀,以示区别。(详见Algorithm::create())
- 通过参数名来设置或检索参数。Algorithm可以通过你指定参数名字符串来提供这样的功能。(详见Algorithm::set()、Algorithm::get())
- 从XML或YAML文件中读写参数。每个算法都可以将参数写到文件中,并且在需要时读出参数。
FaceRecognizer支持的接口
- FaceRecognizer::train()用于将人脸图像数据进行训练
- 根据一张输入的人脸图片进行预测
- 从指定XML或YAML文件中上载或存储模型参数
设置阈值
人脸识别的一个常用情景是待识别人脸是不是输入训练数据集。这个阈值的设置是在具体的FaceRecognizer的构造器中,并通过从Algorithm继承的set和get方法来使用阈值。
创建模型时设置阈值
// Let‘s say we want to keep 10 Eigenfaces and have a threshold value of 10.0 int num_components = 10; double threshold = 10.0; // Then if you want to have a cv::FaceRecognizer with a confidence threshold, // create the concrete implementation with the appropiate parameters: Ptr<FaceRecognizer> model = createEigenFaceRecognizer(num_components, threshold);
运行程序时获取或设置阈值
// The following line reads the threshold from the Eigenfaces model: double current_threshold = model->getDouble("threshold"); // And this line sets the threshold to 0.0: model->set("threshold", 0.0);
将阈值设置为0.0,如果预测类别号为-1,则说明人脸时未知的,没有在数据库中记录
Mat img = imread("person1/3.jpg", CV_LOAD_IMAGE_GRAYSCALE); // Get a prediction from the model. Note: We‘ve set a threshold of 0.0 above, // since the distance is almost always larger than 0.0, you‘ll get -1 as // label, which indicates, this face is unknown int predicted_label = model->predict(img);
FaceRecognizer函数接口
FaceRecognizer::train
函数定义:
void FaceRecognizer::train(InputArrayOfArrays src, InputArray labels) = 0
函数参数为训练的图片,类型为vector<Mat>;图片所对应的标签,类型为vector<int>。
FaceRecognizer::update
函数定义:
void FaceRecognizer::update(InputArrayOfArrays src, InputArray labels)
该方法为更新一个模型(条件是给模型可以被更新,在OpenCV中提供的人脸识别算法中,只有LBP直方图的方法才可以被更新),更新和训练接口的区别是,训练是清空已有的模型,学习一个新的模型,而更新不会删除之前的模型数据。
FaceRecognizer::predict
函数定义:
int FaceRecognizer::predict(InputArray src) const = 0
void FaceRecognizer::predict(InputArray src, int& label, double& confidence) const = 0
第三个参数是对于预测类别的相关置信度。
const后缀表示predict函数不会影响模型内部状态,所以这个方法可以安全被不同线程调用。
下面的小例子是根据训练模型来进行预测。
using namespace cv;
// Do your initialization here (create the cv::FaceRecognizer model) ...
// ...
Mat img = imread("person1/3.jpg", CV_LOAD_IMAGE_GRAYSCALE);
// Some variables for the predicted label and associated confidence (e.g. distance):
int predicted_label = -1;
double predicted_confidence = 0.0;
// Get the prediction and associated confidence from the model
model->predict(img, predicted_label, predicted_confidence);
FaceRecognizer::save
函数定义:
void FaceRecognizer::save(const string& filename) const
void FaceRecognizer::save(FileStorage& fs) const
第一个定义是根据XML或者YAML文件名来存储模型,第二个定义是将模型存成FileStorage实例。
FaceRecognizer::load
函数定义:
void FaceRecognizer::load(const string& filename)
void FaceRecognizer::load(const FileStorage& fs) = 0
这里载入模型参数和save函数的参数意义一样。
建立模型接口
Ptr<FaceRecognizer> createEigenFaceRecognizer(int num_components=0, double threshold=DBL_MAX);
Ptr<FaceRecognizer> createFisherFaceRecognizer(int num_components=0, double threshold=DBL_MAX);
Ptr<FaceRecognizer> createLBPHFaceRecognizer(int radius=1, int neighbors=8, int grid_x=8, int grid_y=8, double threshold=DBL_MAX);
这三个接口分别对应OpenCV的三个人脸识别算法,均返回一个智能指针Ptr类的实例。
小结
这一小节,介绍了OpenCV中人脸识别功能对应的类FaceRecognizer的大体框架,介绍了相关接口和使用方法。在后面的博客,将具体以LBPH算法为例,介绍其原理和匹配源码。
参考资料
OpenCV doc:Algorithm
转载请注明作者Jason Ding及其出处
Github主页(http://jasonding1354.github.io/)
CSDN博客(http://blog.csdn.net/jasonding1354)
简书主页(http://www.jianshu.com/users/2bd9b48f6ea8/latest_articles)
【计算机视觉】OpenCV人脸识别facerec源码分析1——FaceRecognizer概述