mirror of https://github.com/opencv/opencv.git
Open Source Computer Vision Library
https://opencv.org/
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
43 lines
1.0 KiB
43 lines
1.0 KiB
#ifdef HAVE_OPENCV_DNN |
|
typedef dnn::DictValue LayerId; |
|
typedef std::vector<dnn::MatShape> vector_MatShape; |
|
typedef std::vector<std::vector<dnn::MatShape> > vector_vector_MatShape; |
|
|
|
|
|
template<> |
|
bool pyopencv_to(PyObject *o, dnn::DictValue &dv, const char *name) |
|
{ |
|
(void)name; |
|
if (!o || o == Py_None) |
|
return true; //Current state will be used |
|
else if (PyLong_Check(o)) |
|
{ |
|
dv = dnn::DictValue((int64)PyLong_AsLongLong(o)); |
|
return true; |
|
} |
|
else if (PyInt_Check(o)) |
|
{ |
|
dv = dnn::DictValue((int64)PyInt_AS_LONG(o)); |
|
return true; |
|
} |
|
else if (PyFloat_Check(o)) |
|
{ |
|
dv = dnn::DictValue(PyFloat_AS_DOUBLE(o)); |
|
return true; |
|
} |
|
else if (PyString_Check(o)) |
|
{ |
|
dv = dnn::DictValue(String(PyString_AsString(o))); |
|
return true; |
|
} |
|
else |
|
return false; |
|
} |
|
|
|
template<> |
|
bool pyopencv_to(PyObject *o, std::vector<Mat> &blobs, const char *name) //required for Layer::blobs RW |
|
{ |
|
return pyopencvVecConverter<Mat>::to(o, blobs, ArgInfo(name, false)); |
|
} |
|
|
|
#endif
|
|
|