mirror of
https://github.com/opencv/opencv.git
synced 2024-12-02 07:39:57 +08:00
45 lines
1.1 KiB
C++
45 lines
1.1 KiB
C++
#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;
|
|
typedef std::vector<size_t> vector_size_t;
|
|
typedef std::vector<std::vector<Mat> > vector_vector_Mat;
|
|
|
|
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
|