Unverified Commit df2a6bcd authored by Yangze Luo's avatar Yangze Luo Committed by GitHub
Browse files

Fix vad to return zero output for zero input (#3685)

When the input is zero Tensor, the result should be empty.
parent d4cf8d54
......@@ -1660,6 +1660,9 @@ def vad(
flushedLen_ns = (measures_len - num_measures_to_flush) * measure_period_ns
break
# end for window
if not has_triggered:
return waveform[..., :0].view(shape[:-1] + torch.Size([0]))
res = waveform[:, pos - samplesLen_ns + flushedLen_ns :]
# unpack batch
return res.view(shape[:-1] + res.shape[-1:])
......@@ -478,3 +478,18 @@ class TransformsTestBase(TestBaseMixin):
self.assertTrue(diff > 0)
else:
self.assertTrue(diff == 0)
@parameterized.expand(
[
((32000,), (0,), 16000),
((1, 32000), (1, 0), 32000),
((2, 44100), (2, 0), 32000),
((2, 2, 44100), (2, 2, 0), 32000),
]
)
def test_vad_on_zero_audio(self, input_shape, output_shape, sample_rate: int):
"""VAD should return zero when input is zero Tensor"""
inpt = torch.zeros(input_shape, dtype=self.dtype, device=self.device)
expected_output = torch.zeros(output_shape, dtype=self.dtype, device=self.device)
result = T.Vad(sample_rate)(inpt)
self.assertEqual(result, expected_output)
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment