(self, log_mel_spec)
| 54 | return log_mel_spec, stft |
| 55 | |
| 56 | def _pad_spec(self, log_mel_spec): |
| 57 | n_frames = log_mel_spec.shape[0] |
| 58 | p = self.target_length - n_frames |
| 59 | if p > 0: |
| 60 | m = torch.nn.ZeroPad2d((0, 0, 0, p)) |
| 61 | log_mel_spec = m(log_mel_spec) |
| 62 | elif p < 0: |
| 63 | log_mel_spec = log_mel_spec[: self.target_length, :] |
| 64 | if log_mel_spec.size(-1) % 2 != 0: |
| 65 | log_mel_spec = log_mel_spec[..., :-1] |
| 66 | return log_mel_spec |
| 67 | |
| 68 | def load_full_audio(self, filename): |
| 69 | waveform, sr = torchaudio.load(filename) |
no test coverage detected