diff options
| author | Jean-Marc Valin <jmvalin@amazon.com> | 2023-09-27 20:01:21 +0300 |
|---|---|---|
| committer | Jean-Marc Valin <jmvalin@amazon.com> | 2023-09-27 20:01:21 +0300 |
| commit | 9a7bb764d40b9bb65131b4fa64d88e6b16a4d844 (patch) | |
| tree | c5e16d485a86207aadea9550899e6ef9bc9430a0 | |
| parent | a6b4fe375a22976f5897b189c51e0f42cc842f7b (diff) | |
No features skip needed to align pitch features
| -rw-r--r-- | dnn/torch/neural-pitch/models.py | 3 |
1 files changed, 0 insertions, 3 deletions
diff --git a/dnn/torch/neural-pitch/models.py b/dnn/torch/neural-pitch/models.py index a3819785..6822353e 100644 --- a/dnn/torch/neural-pitch/models.py +++ b/dnn/torch/neural-pitch/models.py @@ -189,9 +189,6 @@ class loader(torch.utils.data.Dataset): class loader_joint(torch.utils.data.Dataset): def __init__(self, features, file_pitch, confidence_threshold = 0.4,context = 100, choice_data = 'both'): self.feat = np.memmap(features, mode='r', dtype=np.int8).reshape(-1,312) - #Skip first first two frames for dump_data to sync with CREPE - self.feat = self.feat[2:,:] - self.xcorr = self.feat[:,:224] self.if_feat = self.feat[:,224:] ground_truth = np.memmap(file_pitch, mode='r', dtype=np.float32).reshape(-1,2) |