Skip to content

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

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