Added set_dnn_prefer_smallest_algorithms() (#1128)

* Added set_dnn_prefer_smallest_algorithms()

* Added description for set_dnn_prefer_smallest_algorithms();w

* Changed architectures to algorithms.
This commit is contained in:
HarveyBrezinaConniffe 2018-02-13 21:28:46 +00:00 committed by Davis E. King
parent ef4b7a0ef2
commit 46e85e45c6

View File

@ -155,6 +155,7 @@ void bind_cnn_face_detection(py::module& m)
"takes a list of images as input returning a 2d list of mmod rectangles"
);
}
m.def("set_dnn_prefer_smallest_algorithms", &set_dnn_prefer_smallest_algorithms, "Tells cuDNN to use slower algorithms that use less RAM.");
{
typedef mmod_rect type;
py::class_<type>(m, "mmod_rectangle", "Wrapper around a rectangle object and a detection confidence score.")