mirrors/kaldifeat
1
0
Fork 0
mirror of https://github.com/csukuangfj/kaldifeat.git synced 2025-08-16 04:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add pickle support to MelBanksOptions.

Browse Source
This commit is contained in:
Fangjun Kuang 2021-11-04 10:40:58 +08:00
parent e5dd272963
commit 6eb7a3b243
3 changed files with 27 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
kaldifeat/python/csrc/feature-window.cc
View File

@ -40,10 +40,11 @@ static void PybindFrameExtractionOptions(py::module &m) {
#endif #endif
.def("__str__", .def("__str__",
[](const PyClass &self) -> std::string { return self.ToString(); }) [](const PyClass &self) -> std::string { return self.ToString(); })
.def(py::pickle([](const PyClass &self) { return AsDict(self); }, .def(py::pickle(
[](py::dict dict) -> PyClass { [](const PyClass &self) -> py::dict { return AsDict(self); },
return FrameExtractionOptionsFromDict(dict); [](py::dict dict) -> PyClass {
})); return FrameExtractionOptionsFromDict(dict);
}));
m.def("num_frames", &NumFrames, py::arg("num_samples"), py::arg("opts"), m.def("num_frames", &NumFrames, py::arg("num_samples"), py::arg("opts"),
py::arg("flush") = true); py::arg("flush") = true);

12
kaldifeat/python/csrc/mel-computations.cc
View File

@ -26,9 +26,15 @@ static void PybindMelBanksOptions(py::module &m) {
[](const PyClass &self) -> std::string { return self.ToString(); }) [](const PyClass &self) -> std::string { return self.ToString(); })
.def("as_dict", .def("as_dict",
[](const PyClass &self) -> py::dict { return AsDict(self); }) [](const PyClass &self) -> py::dict { return AsDict(self); })
.def_static("from_dict", [](py::dict dict) -> PyClass { .def_static("from_dict",
return MelBanksOptionsFromDict(dict); [](py::dict dict) -> PyClass {
}); return MelBanksOptionsFromDict(dict);
})
.def(py::pickle(
[](const PyClass &self) -> py::dict { return AsDict(self); },
[](py::dict dict) -> PyClass {
return MelBanksOptionsFromDict(dict);
}));
} }
void PybindMelComputations(py::module &m) { PybindMelBanksOptions(m); } void PybindMelComputations(py::module &m) { PybindMelBanksOptions(m); }

13
kaldifeat/python/tests/test_mel_bank_options.py
View File

@ -2,6 +2,8 @@
# #
# Copyright (c) 2021 Xiaomi Corporation (authors: Fangjun Kuang) # Copyright (c) 2021 Xiaomi Corporation (authors: Fangjun Kuang)
import pickle
import kaldifeat import kaldifeat
@ -82,12 +84,23 @@ def test_from_dict_full_and_as_dict():
assert opts3.htk_mode is True assert opts3.htk_mode is True
def test_pickle():
opts = kaldifeat.MelBanksOptions()
opts.num_bins = 100
opts.low_freq = 22
data = pickle.dumps(opts)
opts2 = pickle.loads(data)
assert str(opts) == str(opts2)
def main(): def main():
test_default() test_default()
test_set_get() test_set_get()
test_from_empty_dict() test_from_empty_dict()
test_from_dict_partial() test_from_dict_partial()
test_from_dict_full_and_as_dict() test_from_dict_full_and_as_dict()
test_pickle()
if __name__ == "__main__": if __name__ == "__main__":