Skip to content

GitLab

Commit 32ad0ffa authored by Patrick Snape's avatar Patrick Snape
Browse files

Make shape predictor pickleable 

Fix typo as well
parent 60318cb3
Show 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.")
- box is the bounding box to begin the shape prediction inside. \n\
ensures \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,
(arg("images"), arg("object_detections"), arg("detector_filename"), arg("options")),
(arg("images"), arg("object_detections"), arg("predictor_output_filename"), arg("options")),
"requires \n\
- options.lambda > 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