GokseninYuksel commited on
Commit
cb8de63
·
verified ·
1 Parent(s): 92d37f3

Upload feature extractor

Browse files
feature_extraction_gramt_ambisonics.py CHANGED
@@ -177,6 +177,7 @@ class AmbisonicsFeatureExtractor(SequenceFeatureExtractor):
177
  if waveform.shape[0] == 1:
178
  waveform = torch.cat([waveform, waveform, waveform, waveform], dim = 0).unsqueeze(0)
179
  log_mel = melspec(waveform).transpose(3, 2)[0]
 
180
  elif waveform.shape[0] == 2:
181
  waveform = waveform[0]
182
  waveform = torch.cat([waveform, waveform, waveform, waveform], dim = 0).unsqueeze(0)
 
177
  if waveform.shape[0] == 1:
178
  waveform = torch.cat([waveform, waveform, waveform, waveform], dim = 0).unsqueeze(0)
179
  log_mel = melspec(waveform).transpose(3, 2)[0]
180
+ return log_mel
181
  elif waveform.shape[0] == 2:
182
  waveform = waveform[0]
183
  waveform = torch.cat([waveform, waveform, waveform, waveform], dim = 0).unsqueeze(0)