NextCloud/dlib
1
0
Fork 0
mirror of https://github.com/davisking/dlib.git synced 2024-11-01 10:14:53 +08:00
Code Issues Packages Projects Releases Wiki Activity

Made the numpy_image capable of binding to tuples that contain a valid image as

Browse Source 
their first element.
This commit is contained in:
Davis King 2018-05-20 18:41:23 -04:00
parent 1619725f22
commit f917f5e5f7
1 changed files with 9 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
dlib/python/numpy_image.h
View File

@ -358,6 +358,15 @@ namespace pybind11
using type = dlib::numpy_image<pixel_type>; using type = dlib::numpy_image<pixel_type>;
bool load(handle src, bool convert) { bool load(handle src, bool convert) {
// If passed a tuple where the first element of the tuple is a valid
// numpy_image then bind the numpy_image to that element of the tuple.
// We do this because there is a pattern of returning an image and some
// associated metadata. This allows the returned tuple from such functions
// to also be treated as an image without needing to unpack the first
// argument.
if (PyTuple_Check(src.ptr()) && PyTuple_Size(src.ptr()) >= 1)
src = reinterpret_borrow<py::tuple>(src)[0];
if (!type::check_(src)) if (!type::check_(src))
return false; return false;
// stash the output of ensure into a temp variable since assigning it to // stash the output of ensure into a temp variable since assigning it to