Skip to content

GitLab

"git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "4c20624cc66cddc0f7bf9133414d5c85f64cc7b5"
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.") ...@@ -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