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.
35 lines
824 B
35 lines
824 B
#ifdef HAVE_OPENCV_DNN |
|
typedef dnn::DictValue LayerId; |
|
|
|
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 (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
|
|
|