Skip to content

GitLab

Commit 0db8713a authored by visionworkz's avatar visionworkz Committed by Davis E. King
Browse files

Fixed reference count issue (#1222)

parent d36c6619
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 4 deletions
tools/python/src/numpy_returns.cpp
View file @ 0db8713a
...@@ -44,7 +44,7 @@ py::list get_jitter_images(py::object img, size_t num_jitters = 1, bool disturb_ ...@@ -44,7 +44,7 @@ py::list get_jitter_images(py::object img, size_t num_jitters = 1, bool disturb_
npy_uint8 *outdata = (npy_uint8 *) PyArray_DATA((PyArrayObject*) arr); npy_uint8 *outdata = (npy_uint8 *) PyArray_DATA((PyArrayObject*) arr);
memcpy(outdata, image_data(crop), rows * width_step(crop)); memcpy(outdata, image_data(crop), rows * width_step(crop));
py::handle handle(arr); py::handle handle = arr;
// Append image to jittered image list // Append image to jittered image list
jitter_list.append(handle); jitter_list.append(handle);
} }
...@@ -87,7 +87,7 @@ py::list get_face_chips ( ...@@ -87,7 +87,7 @@ py::list get_face_chips (
PyObject *arr = PyArray_SimpleNew(3, dims, NPY_UINT8); PyObject *arr = PyArray_SimpleNew(3, dims, NPY_UINT8);
npy_uint8 *outdata = (npy_uint8 *) PyArray_DATA((PyArrayObject*) arr); npy_uint8 *outdata = (npy_uint8 *) PyArray_DATA((PyArrayObject*) arr);
memcpy(outdata, image_data(chip), rows * width_step(chip)); memcpy(outdata, image_data(chip), rows * width_step(chip));
py::handle handle(arr); py::handle handle = arr;
// Append image to chips list // Append image to chips list
chips_list.append(handle); chips_list.append(handle);
...@@ -114,8 +114,7 @@ py::object get_face_chip ( ...@@ -114,8 +114,7 @@ py::object get_face_chip (
PyObject *arr = PyArray_SimpleNew(3, dims, NPY_UINT8); PyObject *arr = PyArray_SimpleNew(3, dims, NPY_UINT8);
npy_uint8 *outdata = (npy_uint8 *) PyArray_DATA((PyArrayObject *) arr); npy_uint8 *outdata = (npy_uint8 *) PyArray_DATA((PyArrayObject *) arr);
memcpy(outdata, image_data(chip), num_rows(chip) * width_step(chip)); memcpy(outdata, image_data(chip), num_rows(chip) * width_step(chip));
py::handle handle(arr); return py::reinterpret_steal<py::object>(arr);
return handle.cast<py::object>();
} }
// ---------------------------------------------------------------------------------------- // ----------------------------------------------------------------------------------------
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment