Skip to content

GitLab

You need to sign in or sign up before continuing.
Commit 32ad0ffa authored by Patrick Snape's avatar Patrick Snape
Browse files

Make shape predictor pickleable 

Fix typo as well
parent 60318cb3
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 2 deletions
tools/python/src/shape_predictor.cpp
View file @ 32ad0ffa
...@@ -225,11 +225,12 @@ train_shape_predictor() routine.") ...@@ -225,11 +225,12 @@ train_shape_predictor() routine.")
- box is the bounding box to begin the shape prediction inside. \n\ - box is the bounding box to begin the shape prediction inside. \n\
ensures \n\ ensures \n\
- This function runs the shape predictor on the input image and returns \n\ - This function runs the shape predictor on the input image and returns \n\
a single full object detection."); a single full object detection.")
.def_pickle(serialize_pickle<type>());
} }
{ {
def("train_shape_predictor", train_shape_predictor_on_images_py, def("train_shape_predictor", train_shape_predictor_on_images_py,
(arg("images"), arg("object_detections"), arg("detector_filename"), arg("options")), (arg("images"), arg("object_detections"), arg("predictor_output_filename"), arg("options")),
"requires \n\ "requires \n\
- options.lambda > 0 \n\ - options.lambda > 0 \n\
- options.nu > 0 \n\ - options.nu > 0 \n\
......
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