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

Add support for matrix serialization to python API (#2241) 

* Add support for matrix serialization to python API

* add double to function names
This commit is contained in:
Adrià Arrufat 2020-11-22 07:09:06 +09:00 committed by GitHub
parent d7644ef2b7
commit b6bf8aefee
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 12 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
tools/python/src/matrix.cpp
View File

@ -108,6 +108,16 @@ long matrix_double__len__(matrix<double>& c)
return c.nr(); return c.nr();
} }
void matrix_double_serialize(const matrix<double>& m, const std::string& file)
{
serialize(file) << m;
}
void matrix_double_deserialize(matrix<double>& m, const std::string& file)
{
deserialize(file) >> m;
}
struct mat_row struct mat_row
{ {
mat_row() : data(0),size(0) {} mat_row() : data(0),size(0) {}
@ -202,6 +212,8 @@ void bind_matrix(py::module& m)
.def("__str__", &matrix_double__str__) .def("__str__", &matrix_double__str__)
.def("nr", &matrix<double>::nr, "Return the number of rows in the matrix.") .def("nr", &matrix<double>::nr, "Return the number of rows in the matrix.")
.def("nc", &matrix<double>::nc, "Return the number of columns in the matrix.") .def("nc", &matrix<double>::nc, "Return the number of columns in the matrix.")
.def("serialize", &matrix_double_serialize, py::arg("file"), "Serialize the matrix to a file")
.def("deserialize", &matrix_double_deserialize, py::arg("file"), "Deserialize the matrix from a file")
.def("__len__", &matrix_double__len__) .def("__len__", &matrix_double__len__)
.def("__getitem__", &matrix_double__getitem__, py::keep_alive<0,1>()) .def("__getitem__", &matrix_double__getitem__, py::keep_alive<0,1>())
.def_property_readonly("shape", &get_matrix_size) .def_property_readonly("shape", &get_matrix_size)