OrgLion-ML
/
dlib
mirror of https://github.com/davisking/dlib.git
1
0
Fork 0
Code Issues Releases Wiki Activity

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
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
tools/python/src/cnn_face_detector.cpp
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.")