Gave the image_window the ability to display fhog object detectors.

This commit is contained in:
Davis King 2014-02-23 15:24:22 -05:00
parent ddc44067b4
commit 8f6b76e7ca
1 changed files with 20 additions and 0 deletions

View File

@ -155,6 +155,22 @@ void deserialize (simple_object_detector_py& item, std::istream& in)
// ---------------------------------------------------------------------------------------- // ----------------------------------------------------------------------------------------
void image_window_set_image_fhog_detector (
image_window& win,
const frontal_face_detector& det
)
{
win.set_image(draw_fhog(det));
}
void image_window_set_image_simple_detector (
image_window& win,
const simple_object_detector_py& det
)
{
win.set_image(draw_fhog(det.detector));
}
void image_window_set_image ( void image_window_set_image (
image_window& win, image_window& win,
object img object img
@ -358,6 +374,10 @@ ensures \n\
"Create an image window that displays the given numpy image and also has the given title.") "Create an image window that displays the given numpy image and also has the given title.")
.def("set_image", image_window_set_image, arg("image"), .def("set_image", image_window_set_image, arg("image"),
"Make the image_window display the given image.") "Make the image_window display the given image.")
.def("set_image", image_window_set_image_fhog_detector, arg("detector"),
"Make the image_window display the given HOG detector's filters.")
.def("set_image", image_window_set_image_simple_detector, arg("detector"),
"Make the image_window display the given HOG detector's filters.")
.def("set_title", (set_title_funct)&type::set_title, arg("title"), .def("set_title", (set_title_funct)&type::set_title, arg("title"),
"Set the title of the window to the given value.") "Set the title of the window to the given value.")
.def("clear_overlay", &type::clear_overlay, "Remove all overlays from the image_window.") .def("clear_overlay", &type::clear_overlay, "Remove all overlays from the image_window.")