(self, log_mel_spec)
| 78 | return log_mel_spec, stft |
| 79 | |
| 80 | def _pad_spec(self, log_mel_spec): |
| 81 | n_frames = log_mel_spec.shape[0] |
| 82 | p = self.target_length - n_frames |
| 83 | if p > 0: |
| 84 | m = torch.nn.ZeroPad2d((0, 0, 0, p)) |
| 85 | log_mel_spec = m(log_mel_spec) |
| 86 | elif p < 0: |
| 87 | log_mel_spec = log_mel_spec[:self.target_length, :] |
| 88 | if log_mel_spec.size(-1) % 2 != 0: |
| 89 | log_mel_spec = log_mel_spec[..., :-1] |
| 90 | return log_mel_spec |
| 91 | |
| 92 | def load_full_audio(self, filename): |
| 93 | waveform, sr = torchaudio.load(filename) |
no test coverage detected