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

Added bottom_up_cluster to python api (#2739) 

* Added bottom_up_cluster to python api

* Update tools/python/src/face_recognition.cpp

Co-authored-by: Adrià Arrufat <1671644+arrufat@users.noreply.github.com>

* Update tools/python/src/face_recognition.cpp

Co-authored-by: Adrià Arrufat <1671644+arrufat@users.noreply.github.com>

---------

Co-authored-by: Davis E. King <davis685@gmail.com>
Co-authored-by: Adrià Arrufat <1671644+arrufat@users.noreply.github.com>
This commit is contained in:
cchadowitz-pf 2023-03-11 08:16:48 -05:00 committed by GitHub
parent 398e2cb5be
commit 3c0b3620af
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 33 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

33
tools/python/src/face_recognition.cpp
View File

@ -230,6 +230,37 @@ private:
// ----------------------------------------------------------------------------------------
py::list bottom_up_clustering(py::list descriptors, const int min_num_clusters, const double max_dist)
{
DLIB_CASSERT(min_num_clusters > 0);
size_t num_descriptors = py::len(descriptors);
matrix<float> dist(static_cast<long>(num_descriptors), static_cast<long>(num_descriptors));
for (size_t i = 0; i < num_descriptors; ++i)
{
for (size_t j = i+1; j < num_descriptors; ++j)
{
const long i2 = static_cast<long>(i);
const long j2 = static_cast<long>(j);
matrix<double,0,1>& first_descriptor = descriptors[i].cast<matrix<double,0,1>&>();
matrix<double,0,1>& second_descriptor = descriptors[j].cast<matrix<double,0,1>&>();
dist(i2, j2) = dlib::length( first_descriptor- second_descriptor);
dist(j2, i2) = dist(i2, j2);
}
}
std::vector<unsigned long> labels;
const auto num_clusters = dlib::bottom_up_cluster(dist, labels, min_num_clusters, max_dist);
py::list clusters;
for (size_t i = 0; i < labels.size(); ++i)
{
clusters.append(labels[i]);
}
return clusters;
}
py::list chinese_whispers_clustering(py::list descriptors, float threshold)
{
DLIB_CASSERT(threshold > 0);
@ -391,6 +422,8 @@ void bind_face_recognition(py::module &m)
"Takes an image and a full_object_detections object that reference faces in that image and saves the faces with the specified file name prefix. The faces will be rotated upright and scaled to 150x150 pixels or with the optional specified size and padding.",
py::arg("img"), py::arg("faces"), py::arg("chip_filename"), py::arg("size")=150, py::arg("padding")=0.25
);
m.def("bottom_up_clustering", &bottom_up_clustering, py::arg("descriptors"), py::arg("min_num_clusters")=1, py::arg("max_dist")=0.6,
"Takes a list of descriptors and returns a list that contains a label for each descriptor. Clustering is done using dlib::bottom_up_cluster.");
m.def("chinese_whispers_clustering", &chinese_whispers_clustering, py::arg("descriptors"), py::arg("threshold"),
"Takes a list of descriptors and returns a list that contains a label for each descriptor. Clustering is done using dlib::chinese_whispers."
);