We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 86c40b8 commit 78bbf70Copy full SHA for 78bbf70
src/torchaudio/utils/__init__.py
@@ -4,14 +4,17 @@
4
from .download import download_asset
5
import os
6
7
-def load_torchcodec(file, **args):
+def load_torchcodec(file, normalize=True, channels_first=True, **args):
8
+ if not normalize:
9
+ raise Exception("Torchcodec does not support non-normalized file reading")
10
from torchcodec.decoders import AudioDecoder
11
decoder = AudioDecoder(file)
12
if 'start_seconds' in args or 'stop_seconds' in args:
13
samples = decoder.get_samples_played_in_range(**args)
14
else:
15
samples = decoder.get_all_samples()
- return (samples.data, samples.sample_rate)
16
+ data = samples.data if channels_first else samples.data.T
17
+ return (data, samples.sample_rate)
18
19
__all__ = [
20
"load_torchcodec",
0 commit comments