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

Add TODO task lists.

Browse Source
This commit is contained in:
Fangjun Kuang 2021-02-27 00:09:36 +08:00
parent ded479fc10
commit 7c8ef68bb5
2 changed files with 17 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
README.md
View File

@ -9,3 +9,15 @@ Wrap kaldi's feature computations to Python with PyTorch support.
```bash ```bash
pip install kaldifeat pip install kaldifeat
``` ```
# TODOs
- [ ] Add Python interface
- [ ] Support torch.device so that it can switch between CUDA and CPU
- [ ] Add unit tests
- [ ] Set up GitHub actions
- [ ] Benchmark its speed and compare it with Kaldi
- [ ] Support batch processing of multiple waves
- [ ] Handle non-default parameters
- [ ] Support MFCC and other features available in Kaldi
- [ ] Publish it to PyPI

5
kaldifeat/csrc/feature-window.cc
View File

@ -134,7 +134,12 @@ torch::Tensor Dither(const torch::Tensor &wave, float dither_value) {
if (dither_value == 0.0f) wave; if (dither_value == 0.0f) wave;
torch::Tensor rand_gauss = torch::randn_like(wave); torch::Tensor rand_gauss = torch::randn_like(wave);
#if 1
return wave + rand_gauss * dither_value; return wave + rand_gauss * dither_value;
#else
// use in-place version of wave and change its to pointer type
wave_->add_(rand_gauss, dither_value);
#endif
} }
void Preemphasize(float preemph_coeff, torch::Tensor *wave) { void Preemphasize(float preemph_coeff, torch::Tensor *wave) {