OrgLion-ML
/
dlib
mirror of https://github.com/davisking/dlib.git
1
0
Fork 0
Code Issues Releases Wiki Activity

Expose get_face_chip_details to Python (#2238)

This commit is contained in:
Frankie Robertson 2020-11-22 00:07:28 +02:00 committed by GitHub
parent 96a75568be
commit d7644ef2b7
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 18 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
tools/python/src/image2.cpp
View File

@ -372,6 +372,11 @@ ensures \n\
.def_readwrite("rows", &chip_details::rows) .def_readwrite("rows", &chip_details::rows)
.def_readwrite("cols", &chip_details::cols); .def_readwrite("cols", &chip_details::cols);
{
typedef std::vector<chip_details> type;
py::bind_vector<type>(m, "chip_detailss", "An array of chip_details objects.")
.def("extend", extend_vector_with_python_list<type>);
}
m.def("extract_image_chip", &py_extract_image_chip<uint8_t>, py::arg("img"), py::arg("chip_location")); m.def("extract_image_chip", &py_extract_image_chip<uint8_t>, py::arg("img"), py::arg("chip_location"));
m.def("extract_image_chip", &py_extract_image_chip<uint16_t>, py::arg("img"), py::arg("chip_location")); m.def("extract_image_chip", &py_extract_image_chip<uint16_t>, py::arg("img"), py::arg("chip_location"));
@ -447,6 +452,19 @@ ensures \n\
!*/ !*/
); );
m.def("get_face_chip_details",
static_cast<chip_details (*)(const full_object_detection&, const unsigned long, const double)>(&get_face_chip_details),
py::arg("det"), py::arg("size")=200, py::arg("padding")=0.2,
"Given a full_object_detection det, returns a chip_details object which can be \n\
used to extract an image of given size and padding."
);
m.def("get_face_chip_details",
static_cast<std::vector<chip_details> (*)(const std::vector<full_object_detection>&, const unsigned long, const double)>(&get_face_chip_details),
py::arg("dets"), py::arg("size")=200, py::arg("padding")=0.2,
"Given a list of full_object_detection dets, returns a chip_details object which can be \n\
used to extract an image of given size and padding."
);
} }
// ---------------------------------------------------------------------------------------- // ----------------------------------------------------------------------------------------