(self, waveform)
| 42 | ) |
| 43 | |
| 44 | def wav_feature_extraction(self, waveform): |
| 45 | import utilities.audio as Audio |
| 46 | |
| 47 | waveform = waveform[0, ...] |
| 48 | waveform = torch.FloatTensor(waveform) |
| 49 | log_mel_spec, stft, _ = Audio.tools.get_mel_from_wav(waveform, self.STFT) |
| 50 | log_mel_spec = torch.FloatTensor(log_mel_spec.T) |
| 51 | stft = torch.FloatTensor(stft.T) |
| 52 | log_mel_spec = self._pad_spec(log_mel_spec) |
| 53 | stft = self._pad_spec(stft) |
| 54 | return log_mel_spec, stft |
| 55 | |
| 56 | def _pad_spec(self, log_mel_spec): |
| 57 | n_frames = log_mel_spec.shape[0] |
no test coverage detected