kotovalexarian-likes-github/ruby-opencv
1
0
Fork 0
mirror of https://github.com/ruby-opencv/ruby-opencv synced 2023-03-27 23:22:12 -04:00
Code Releases Activity

add FaceRecognizer class and modify EigenFaces to inherit it

Browse source
This commit is contained in:
ser1zw 2013-06-23 19:56:20 +09:00
parent 6938b4430c
commit 5b57e355f9
6 changed files with 193 additions and 121 deletions
Download patch file Download diff file Expand all files Collapse all files

109
ext/opencv/eigenfaces.cpp
View file

@ -61,109 +61,6 @@ rb_initialize(int argc, VALUE argv[], VALUE self)
return self;
}
/*
* call-seq:
* train(src, labels)
*
* Trains a FaceRecognizer with given data and associated labels.
*/
VALUE
rb_train(VALUE self, VALUE src, VALUE labels)
{
Check_Type(src, T_ARRAY);
Check_Type(labels, T_ARRAY);
VALUE *src_ptr = RARRAY_PTR(src);
int src_size = RARRAY_LEN(src);
std::vector<cv::Mat> images;
for (int i = 0; i < src_size; i++) {
images.push_back(cv::Mat(CVMAT_WITH_CHECK(src_ptr[i])));
}
VALUE *labels_ptr = RARRAY_PTR(labels);
int labels_size = RARRAY_LEN(labels);
std::vector<int> local_labels;
for (int i = 0; i < labels_size; i++) {
local_labels.push_back(NUM2INT(labels_ptr[i]));
}
cv::FaceRecognizer *self_ptr = FACERECOGNIZER(self);
try {
self_ptr->train(images, local_labels);
}
catch (cv::Exception& e) {
raise_cverror(e);
}
return Qnil;
}
/*
* call-seq:
* predict(src)
*
* Predicts a label and associated confidence (e.g. distance) for a given input image.
*/
VALUE
rb_predict(VALUE self, VALUE src)
{
cv::Mat mat = cv::Mat(CVMAT_WITH_CHECK(src));
cv::FaceRecognizer *self_ptr = FACERECOGNIZER(self);
int label;
try {
label = self_ptr->predict(mat);
}
catch (cv::Exception& e) {
raise_cverror(e);
}
return INT2NUM(label);
}
/*
* call-seq:
* save(filename)
*
* Saves this model to a given filename, either as XML or YAML.
*/
VALUE
rb_save(VALUE self, VALUE filename)
{
Check_Type(filename, T_STRING);
cv::FaceRecognizer *self_ptr = FACERECOGNIZER(self);
try {
char* s = StringValueCStr(filename);
self_ptr->save(std::string(s));
}
catch (cv::Exception& e) {
raise_cverror(e);
}
return Qnil;
}
/*
* call-seq:
* load(filename)
*
* Loads a FaceRecognizer and its model state.
*/
VALUE
rb_load(VALUE self, VALUE filename)
{
Check_Type(filename, T_STRING);
cv::FaceRecognizer *self_ptr = FACERECOGNIZER(self);
try {
char* s = StringValueCStr(filename);
self_ptr->load(std::string(s));
}
catch (cv::Exception& e) {
raise_cverror(e);
}
return Qnil;
}
void
define_ruby_class()
{
@ -175,13 +72,9 @@ define_ruby_class()
* note: this comment is used by rdoc.
*/
VALUE opencv = rb_module_opencv();
rb_klass = rb_define_class_under(opencv, "EigenFaces", rb_cObject);
rb_klass = rb_define_class_under(opencv, "EigenFaces", cFaceRecognizer::rb_class());
rb_define_alloc_func(rb_klass, rb_allocate);
rb_define_private_method(rb_klass, "initialize", RUBY_METHOD_FUNC(rb_initialize), -1);
rb_define_method(rb_klass, "train", RUBY_METHOD_FUNC(rb_train), 2);
rb_define_method(rb_klass, "predict", RUBY_METHOD_FUNC(rb_predict), 1);
rb_define_method(rb_klass, "save", RUBY_METHOD_FUNC(rb_save), 1);
rb_define_method(rb_klass, "load", RUBY_METHOD_FUNC(rb_load), 1);
}
__NAMESPACE_END_EIGENFACES

13
ext/opencv/eigenfaces.h
View file

@ -23,22 +23,9 @@ VALUE rb_class();
void define_ruby_class();
VALUE rb_allocate(VALUE klass);
VALUE rb_initialize(int argc, VALUE argv[], VALUE self);
VALUE rb_train(VALUE self, VALUE src, VALUE labels);
VALUE rb_save(VALUE self, VALUE filename);
VALUE rb_load(VALUE self, VALUE filename);
__NAMESPACE_END_EIGENFACES
inline cv::FaceRecognizer*
FACERECOGNIZER(VALUE object)
{
cv::FaceRecognizer *ptr;
Data_Get_Struct(object, cv::FaceRecognizer, ptr);
return ptr;
}
__NAMESPACE_END_OPENCV
#endif // RUBY_OPENCV_EIGENFACES_H

149
ext/opencv/facerecognizer.cpp Normal file
View file

@ -0,0 +1,149 @@
/************************************************************
facerecognizer.cpp -
$Author: ser1zw $
Copyright (C) 2013 ser1zw
************************************************************/
#include <stdio.h>
#include "facerecognizer.h"
/*
* Document-class: OpenCV::FaceRecognizer
*
*/
__NAMESPACE_BEGIN_OPENCV
__NAMESPACE_BEGIN_FACERECOGNIZER
VALUE rb_klass;
VALUE
rb_class()
{
return rb_klass;
}
/*
* call-seq:
* train(src, labels)
*
* Trains a FaceRecognizer with given data and associated labels.
*/
VALUE
rb_train(VALUE self, VALUE src, VALUE labels)
{
Check_Type(src, T_ARRAY);
Check_Type(labels, T_ARRAY);
VALUE *src_ptr = RARRAY_PTR(src);
int src_size = RARRAY_LEN(src);
std::vector<cv::Mat> images;
for (int i = 0; i < src_size; i++) {
images.push_back(cv::Mat(CVMAT_WITH_CHECK(src_ptr[i])));
}
VALUE *labels_ptr = RARRAY_PTR(labels);
int labels_size = RARRAY_LEN(labels);
std::vector<int> local_labels;
for (int i = 0; i < labels_size; i++) {
local_labels.push_back(NUM2INT(labels_ptr[i]));
}
cv::FaceRecognizer *self_ptr = FACERECOGNIZER(self);
try {
self_ptr->train(images, local_labels);
}
catch (cv::Exception& e) {
raise_cverror(e);
}
return Qnil;
}
/*
* call-seq:
* predict(src)
*
* Predicts a label and associated confidence (e.g. distance) for a given input image.
*/
VALUE
rb_predict(VALUE self, VALUE src)
{
cv::Mat mat = cv::Mat(CVMAT_WITH_CHECK(src));
cv::FaceRecognizer *self_ptr = FACERECOGNIZER(self);
int label;
try {
label = self_ptr->predict(mat);
}
catch (cv::Exception& e) {
raise_cverror(e);
}
return INT2NUM(label);
}
/*
* call-seq:
* save(filename)
*
* Saves this model to a given filename, either as XML or YAML.
*/
VALUE
rb_save(VALUE self, VALUE filename)
{
Check_Type(filename, T_STRING);
cv::FaceRecognizer *self_ptr = FACERECOGNIZER(self);
try {
char* s = StringValueCStr(filename);
self_ptr->save(std::string(s));
}
catch (cv::Exception& e) {
raise_cverror(e);
}
return Qnil;
}
/*
* call-seq:
* load(filename)
*
* Loads a FaceRecognizer and its model state.
*/
VALUE
rb_load(VALUE self, VALUE filename)
{
Check_Type(filename, T_STRING);
cv::FaceRecognizer *self_ptr = FACERECOGNIZER(self);
try {
char* s = StringValueCStr(filename);
self_ptr->load(std::string(s));
}
catch (cv::Exception& e) {
raise_cverror(e);
}
return Qnil;
}
void
define_ruby_class()
{
if (rb_klass)
return;
/*
* opencv = rb_define_module("OpenCV");
*
* note: this comment is used by rdoc.
*/
VALUE opencv = rb_module_opencv();
rb_klass = rb_define_class_under(opencv, "FaceRecognizer", rb_cObject);
rb_define_method(rb_klass, "train", RUBY_METHOD_FUNC(rb_train), 2);
rb_define_method(rb_klass, "predict", RUBY_METHOD_FUNC(rb_predict), 1);
rb_define_method(rb_klass, "save", RUBY_METHOD_FUNC(rb_save), 1);
rb_define_method(rb_klass, "load", RUBY_METHOD_FUNC(rb_load), 1);
}
__NAMESPACE_END_FACERECOGNIZER
__NAMESPACE_END_OPENCV

41
ext/opencv/facerecognizer.h Normal file
View file

@ -0,0 +1,41 @@
/************************************************************
facerecognizer.h
$Author: ser1zw $
Copyright (C) 2013 ser1zw
************************************************************/
#ifndef RUBY_OPENCV_FACERECOGNIZER_H
#define RUBY_OPENCV_FACERECOGNIZER_H
#include "opencv.h"
#define __NAMESPACE_BEGIN_FACERECOGNIZER namespace cFaceRecognizer {
#define __NAMESPACE_END_FACERECOGNIZER }
__NAMESPACE_BEGIN_OPENCV
__NAMESPACE_BEGIN_FACERECOGNIZER
VALUE rb_class();
void define_ruby_class();
VALUE rb_train(VALUE self, VALUE src, VALUE labels);
VALUE rb_save(VALUE self, VALUE filename);
VALUE rb_load(VALUE self, VALUE filename);
__NAMESPACE_END_FACERECOGNIZER
inline cv::FaceRecognizer*
FACERECOGNIZER(VALUE object)
{
cv::FaceRecognizer *ptr;
Data_Get_Struct(object, cv::FaceRecognizer, ptr);
return ptr;
}
__NAMESPACE_END_OPENCV
#endif // RUBY_OPENCV_FACERECOGNIZER_H

1
ext/opencv/opencv.cpp
View file

@ -707,6 +707,7 @@ extern "C" {
mOpenCV::cCvAvgComp::define_ruby_class();
mOpenCV::cCvHaarClassifierCascade::define_ruby_class();
mOpenCV::cFaceRecognizer::define_ruby_class();
mOpenCV::cEigenFaces::define_ruby_class();
mOpenCV::mGUI::define_ruby_module();

1
ext/opencv/opencv.h
View file

@ -130,6 +130,7 @@ extern "C" {
#include "cvfeaturetree.h"
#include "facerecognizer.h"
#include "eigenfaces.h"
// GUI