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)
|