basic dnn support
Esse commit está contido em:
Vladimir Goncharov
@@ -51,6 +51,7 @@ if test "$PHP_OPENCV" != "no"; then
|
||||
source/opencv2/face/opencv_facerec.cc \
|
||||
source/opencv2/face/opencv_facemarkLBF.cc \
|
||||
source/opencv2/core/opencv_cvdef.cc \
|
||||
source/opencv2/dnn/opencv_dnn.cc \
|
||||
source/opencv2/opencv_ml.cc"
|
||||
|
||||
|
||||
|
||||
@@ -43,6 +43,7 @@ extern "C" {
|
||||
#include "source/opencv2/core/opencv_utility.h"
|
||||
#include "source/opencv2/opencv_ml.h"
|
||||
#include "source/opencv2/core/opencv_cvdef.h"
|
||||
#include "source/opencv2/dnn/opencv_dnn.h"
|
||||
|
||||
/* If you declare any globals in php_opencv.h uncomment this:
|
||||
ZEND_DECLARE_MODULE_GLOBALS(opencv)
|
||||
@@ -124,6 +125,7 @@ PHP_MINIT_FUNCTION(opencv)
|
||||
opencv_objdetect_init(module_number);
|
||||
opencv_videoio_init(module_number);
|
||||
opencv_face_init(module_number);
|
||||
opencv_dnn_net_init(module_number);
|
||||
opencv_ml_init(module_number);
|
||||
opencv_cvdef_init(module_number);
|
||||
|
||||
|
||||
@@ -49,6 +49,7 @@ using namespace cv;
|
||||
#define OPENCV_NS "CV"
|
||||
#define OPENCV_FACE_NS ZEND_NS_NAME(OPENCV_NS,"Face")
|
||||
#define OPENCV_ML_NS ZEND_NS_NAME(OPENCV_NS,"ML")
|
||||
#define OPENCV_DNN_NS ZEND_NS_NAME(OPENCV_NS,"DNN")
|
||||
|
||||
#define OPENCV_CONNECT(text1,text2) text1##text2
|
||||
|
||||
|
||||
@@ -0,0 +1,211 @@
|
||||
/*
|
||||
+----------------------------------------------------------------------+
|
||||
| PHP-OpenCV |
|
||||
+----------------------------------------------------------------------+
|
||||
| This source file is subject to version 2.0 of the Apache license, |
|
||||
| that is bundled with this package in the file LICENSE, and is |
|
||||
| available through the world-wide-web at the following url: |
|
||||
| http://www.apache.org/licenses/LICENSE-2.0.html |
|
||||
| If you did not receive a copy of the Apache2.0 license and are unable|
|
||||
| to obtain it through the world-wide-web, please send a note to |
|
||||
| hihozhou@gmail.com so we can mail you a copy immediately. |
|
||||
+----------------------------------------------------------------------+
|
||||
| Author: HaiHao Zhou <hihozhou@gmail.com> |
|
||||
+----------------------------------------------------------------------+
|
||||
*/
|
||||
#include "../../../php_opencv.h"
|
||||
#include "opencv_dnn.h"
|
||||
|
||||
#ifdef HAVE_OPENCV_DNN
|
||||
|
||||
#include "../core/opencv_type.h"
|
||||
#include "../core/opencv_mat.h"
|
||||
#include "../../../opencv_exception.h"
|
||||
|
||||
using namespace dnn;
|
||||
using namespace std;
|
||||
|
||||
#define Z_PHP_DNN_NET_OBJ_P(zv) get_dnn_net_obj(Z_OBJ_P(zv))
|
||||
|
||||
typedef struct _opencv_dnn_net_object{
|
||||
zend_object std;
|
||||
Net DNNNet;
|
||||
}opencv_dnn_net_object;
|
||||
|
||||
|
||||
/**
|
||||
* @param obj
|
||||
* @return
|
||||
*/
|
||||
static inline opencv_dnn_net_object* get_dnn_net_obj(zend_object *obj) {
|
||||
return (opencv_dnn_net_object*)((char*)(obj) - XtOffsetOf(opencv_dnn_net_object, std));
|
||||
}
|
||||
|
||||
zend_object_handlers opencv_dnn_net_object_handlers;
|
||||
|
||||
zend_class_entry *opencv_dnn_net_ce;
|
||||
|
||||
zend_class_entry *opencv_base_dnn_net_ce;
|
||||
|
||||
PHP_METHOD(opencv_dnn_net, blobFromImage)
|
||||
{
|
||||
zval *image_zval, *size_zval, *mean_zval;
|
||||
double scalefactor = 1.;
|
||||
bool swapRB = true, crop = true;
|
||||
|
||||
if (zend_parse_parameters(ZEND_NUM_ARGS(), "OdOO|bb",
|
||||
&image_zval, opencv_mat_ce,
|
||||
&scalefactor,
|
||||
&size_zval, opencv_size_ce,
|
||||
&mean_zval, opencv_scalar_ce,
|
||||
&swapRB,
|
||||
&crop
|
||||
) == FAILURE) {
|
||||
RETURN_NULL();
|
||||
}
|
||||
|
||||
opencv_mat_object *image_object = Z_PHP_MAT_OBJ_P(image_zval);
|
||||
opencv_size_object *size_object = Z_PHP_SIZE_OBJ_P(size_zval);
|
||||
opencv_scalar_object *mean_object = Z_PHP_SCALAR_OBJ_P(mean_zval);
|
||||
|
||||
Mat im = blobFromImage(*image_object->mat, scalefactor, *size_object->size, *mean_object->scalar, swapRB, crop);
|
||||
|
||||
zval instance;
|
||||
object_init_ex(&instance, opencv_mat_ce);
|
||||
opencv_mat_object *new_obj = Z_PHP_MAT_OBJ_P(&instance);
|
||||
new_obj->mat=new Mat(im);
|
||||
opencv_mat_update_property_by_c_mat(&instance, new_obj->mat);
|
||||
RETURN_ZVAL(&instance,0,0); //return php Mat object
|
||||
}
|
||||
|
||||
PHP_METHOD(opencv_dnn_net, readNetFromTorch)
|
||||
{
|
||||
char *filename;
|
||||
size_t filename_len;
|
||||
if (zend_parse_parameters(ZEND_NUM_ARGS(), "s", &filename, &filename_len) == FAILURE) {
|
||||
RETURN_NULL();
|
||||
}
|
||||
|
||||
zval instance;
|
||||
object_init_ex(&instance, opencv_dnn_net_ce);
|
||||
opencv_dnn_net_object *obj = Z_PHP_DNN_NET_OBJ_P(&instance);
|
||||
obj->DNNNet = readNetFromTorch(filename, true);
|
||||
|
||||
RETURN_ZVAL(&instance,0,0);
|
||||
}
|
||||
|
||||
ZEND_BEGIN_ARG_INFO_EX(opencv_dnn_net_readNetFromCaffe_arginfo, 0, 0, 2)
|
||||
ZEND_ARG_INFO(0, prototxt)
|
||||
ZEND_ARG_INFO(0, caffeModel)
|
||||
ZEND_END_ARG_INFO()
|
||||
|
||||
PHP_METHOD(opencv_dnn_net, readNetFromCaffe)
|
||||
{
|
||||
char *prototxt, *caffeModel;
|
||||
size_t prototxt_len, caffeModel_len;
|
||||
if (zend_parse_parameters(ZEND_NUM_ARGS(), "ss", &prototxt, &prototxt_len, &caffeModel, &caffeModel_len) == FAILURE) {
|
||||
RETURN_NULL();
|
||||
}
|
||||
|
||||
ifstream protoFile;
|
||||
protoFile.open(prototxt);//open the input file
|
||||
stringstream protoStream;
|
||||
protoStream << protoFile.rdbuf();//read the file
|
||||
string protoData = protoStream.str();
|
||||
|
||||
ifstream modelFile;
|
||||
modelFile.open(caffeModel, ios::binary);//open the input file
|
||||
stringstream modelStream;
|
||||
modelStream << modelFile.rdbuf();//read the file
|
||||
string modelData = modelStream.str();
|
||||
|
||||
zval instance;
|
||||
object_init_ex(&instance, opencv_dnn_net_ce);
|
||||
opencv_dnn_net_object *obj = Z_PHP_DNN_NET_OBJ_P(&instance);
|
||||
|
||||
//obj->DNNNet = readNetFromCaffe(protoData.c_str(), protoData.size());
|
||||
obj->DNNNet = readNetFromCaffe(protoData.c_str(), protoData.size(), modelData.c_str(), modelData.size());
|
||||
|
||||
RETURN_ZVAL(&instance,0,0);
|
||||
}
|
||||
|
||||
PHP_METHOD(opencv_dnn_net, setInput)
|
||||
{
|
||||
zval *image_zval;
|
||||
char *name;
|
||||
size_t name_len;
|
||||
|
||||
if (zend_parse_parameters(ZEND_NUM_ARGS(), "Os", &image_zval, opencv_mat_ce, &name, &name_len) == FAILURE) {
|
||||
RETURN_NULL();
|
||||
}
|
||||
|
||||
opencv_mat_object *image_object = Z_PHP_MAT_OBJ_P(image_zval);
|
||||
|
||||
opencv_dnn_net_object *obj = Z_PHP_DNN_NET_OBJ_P(getThis());
|
||||
obj->DNNNet.setInput(*(image_object->mat), name);
|
||||
|
||||
RETURN_NULL();
|
||||
}
|
||||
|
||||
PHP_METHOD(opencv_dnn_net, forward)
|
||||
{
|
||||
zval *image_zval;
|
||||
char *name;
|
||||
|
||||
opencv_dnn_net_object *obj = Z_PHP_DNN_NET_OBJ_P(getThis());
|
||||
Mat image = obj->DNNNet.forward();
|
||||
|
||||
zval instance;
|
||||
object_init_ex(&instance, opencv_mat_ce);
|
||||
opencv_mat_object *new_obj = Z_PHP_MAT_OBJ_P(&instance);
|
||||
new_obj->mat=new Mat(image);
|
||||
opencv_mat_update_property_by_c_mat(&instance, new_obj->mat);
|
||||
RETURN_ZVAL(&instance,0,0); //return php Mat object
|
||||
}
|
||||
|
||||
|
||||
/**
|
||||
* opencv_dnn_net_methods[]
|
||||
*/
|
||||
const zend_function_entry opencv_dnn_net_methods[] = {
|
||||
PHP_ME(opencv_dnn_net, blobFromImage, NULL, ZEND_ACC_PUBLIC|ZEND_ACC_STATIC)
|
||||
PHP_ME(opencv_dnn_net, readNetFromTorch, NULL, ZEND_ACC_PUBLIC|ZEND_ACC_STATIC)
|
||||
PHP_ME(opencv_dnn_net, readNetFromCaffe, NULL, ZEND_ACC_PUBLIC|ZEND_ACC_STATIC)
|
||||
PHP_ME(opencv_dnn_net, setInput, NULL, ZEND_ACC_PUBLIC)
|
||||
PHP_ME(opencv_dnn_net, forward, NULL, ZEND_ACC_PUBLIC)
|
||||
PHP_FE_END
|
||||
};
|
||||
/* }}} */
|
||||
|
||||
/**
|
||||
* @param type
|
||||
* @return
|
||||
*/
|
||||
zend_object* opencv_dnn_net_handler(zend_class_entry *type)
|
||||
{
|
||||
size_t size = sizeof(opencv_dnn_net_object);
|
||||
opencv_dnn_net_object *obj = (opencv_dnn_net_object *)ecalloc(1,size);
|
||||
memset(obj, 0, sizeof(opencv_dnn_net_object));
|
||||
zend_object_std_init(&obj->std, type);
|
||||
object_properties_init(&obj->std, type);
|
||||
obj->std.ce = type;
|
||||
obj->std.handlers = &opencv_dnn_net_object_handlers;
|
||||
return &obj->std;
|
||||
}
|
||||
|
||||
void opencv_dnn_net_init(int module_number){
|
||||
zend_class_entry ce;
|
||||
INIT_NS_CLASS_ENTRY(ce,OPENCV_DNN_NS, "Net", opencv_dnn_net_methods);
|
||||
opencv_dnn_net_ce = zend_register_internal_class_ex(&ce, opencv_dnn_net_ce);
|
||||
|
||||
opencv_dnn_net_ce->create_object = opencv_dnn_net_handler;
|
||||
memcpy(&opencv_dnn_net_object_handlers, zend_get_std_object_handlers(), sizeof(zend_object_handlers));
|
||||
}
|
||||
|
||||
#else
|
||||
|
||||
void opencv_dnn_net_init(int module_number){
|
||||
|
||||
}
|
||||
|
||||
#endif
|
||||
@@ -0,0 +1,29 @@
|
||||
/*
|
||||
+----------------------------------------------------------------------+
|
||||
| PHP-OpenCV |
|
||||
+----------------------------------------------------------------------+
|
||||
| This source file is subject to version 2.0 of the Apache license, |
|
||||
| that is bundled with this package in the file LICENSE, and is |
|
||||
| available through the world-wide-web at the following url: |
|
||||
| http://www.apache.org/licenses/LICENSE-2.0.html |
|
||||
| If you did not receive a copy of the Apache2.0 license and are unable|
|
||||
| to obtain it through the world-wide-web, please send a note to |
|
||||
| hihozhou@gmail.com so we can mail you a copy immediately. |
|
||||
+----------------------------------------------------------------------+
|
||||
| Author: HaiHao Zhou <hihozhou@gmail.com> |
|
||||
+----------------------------------------------------------------------+
|
||||
*/
|
||||
|
||||
|
||||
#ifndef PHP_OPENCV_DNN_NET_H
|
||||
#define PHP_OPENCV_DNN_NET_H
|
||||
|
||||
#include <opencv2/dnn.hpp>
|
||||
#include <string>
|
||||
#include <fstream>
|
||||
#include <streambuf>
|
||||
|
||||
|
||||
extern void opencv_dnn_net_init(int module_number);
|
||||
|
||||
#endif
|
||||
Referência em uma Nova Issue
Bloquear um usuário