README.md 4.29 KB
Newer Older
Soumith Chintala's avatar
Soumith Chintala committed
1
torchaudio: an audio library for PyTorch
Vincent QB's avatar
Vincent QB committed
2
========================================
Soumith Chintala's avatar
Soumith Chintala committed
3

moto's avatar
moto committed
4
[![Documentation](https://img.shields.io/badge/dynamic/json.svg?label=docs&url=https%3A%2F%2Fpypi.org%2Fpypi%2Ftorchaudio%2Fjson&query=%24.info.version&colorB=brightgreen&prefix=v)](https://pytorch.org/audio/main/)
moto's avatar
moto committed
5
6
[![Anaconda Badge](https://anaconda.org/pytorch/torchaudio/badges/downloads.svg)](https://anaconda.org/pytorch/torchaudio)
[![Anaconda-Server Badge](https://anaconda.org/pytorch/torchaudio/badges/platforms.svg)](https://anaconda.org/pytorch/torchaudio)
7

moto's avatar
moto committed
8
9
![TorchAudio Logo](docs/source/_static/img/logo.png)

jamarshon's avatar
jamarshon committed
10
The aim of torchaudio is to apply [PyTorch](https://github.com/pytorch/pytorch) to
11
the audio domain. By supporting PyTorch, torchaudio follows the same philosophy
jamarshon's avatar
jamarshon committed
12
13
of providing strong GPU acceleration, having a focus on trainable features through
the autograd system, and having consistent style (tensor names and dimension names).
14
Therefore, it is primarily a machine learning library and not a general signal
15
processing library. The benefits of PyTorch can be seen in torchaudio through
16
having all the computations be through PyTorch operations which makes it easy
jamarshon's avatar
jamarshon committed
17
18
to use and feel like a natural extension.

moto's avatar
moto committed
19
- [Support audio I/O (Load files, Save files)](http://pytorch.org/audio/main/)
moto's avatar
moto committed
20
  - Load a variety of audio formats, such as `wav`, `mp3`, `ogg`, `flac`, `opus`, `sphere`, into a torch Tensor using SoX
moto's avatar
moto committed
21
22
  - [Kaldi (ark/scp)](http://pytorch.org/audio/main/kaldi_io.html)
- [Dataloaders for common audio datasets](http://pytorch.org/audio/main/datasets.html)
Zhaoheng Ni's avatar
Zhaoheng Ni committed
23
24
- Audio and speech processing functions
  - [forced_align](https://pytorch.org/audio/main/generated/torchaudio.functional.forced_align.html)
25
- Common audio transforms
Zhaoheng Ni's avatar
Zhaoheng Ni committed
26
  - [Spectrogram, AmplitudeToDB, MelScale, MelSpectrogram, MFCC, MuLawEncoding, MuLawDecoding, Resample](http://pytorch.org/audio/main/transforms.html)
27
- Compliance interfaces: Run code using PyTorch that align with other libraries
Zhaoheng Ni's avatar
Zhaoheng Ni committed
28
  - [Kaldi: spectrogram, fbank, mfcc](https://pytorch.org/audio/main/compliance.kaldi.html)
Soumith Chintala's avatar
Soumith Chintala committed
29
30
31
32

Installation
------------

33
Please refer to https://pytorch.org/audio/main/installation.html for installation and build process of TorchAudio.
34

35

Soumith Chintala's avatar
Soumith Chintala committed
36
API Reference
Vincent QB's avatar
Vincent QB committed
37
-------------
SeanNaren's avatar
SeanNaren committed
38

moto's avatar
moto committed
39
API Reference is located here: http://pytorch.org/audio/main/
Vincent QB's avatar
Vincent QB committed
40

jamarshon's avatar
jamarshon committed
41
42
43
Contributing Guidelines
-----------------------

Nicolas Hug's avatar
Nicolas Hug committed
44
Please refer to [CONTRIBUTING.md](./CONTRIBUTING.md)
Vincent QB's avatar
Vincent QB committed
45

yangarbiter's avatar
yangarbiter committed
46
47
48
49
50
51
52
53
54
55
56
57
58
59
Citation
--------

If you find this package useful, please cite as:

```bibtex
@article{yang2021torchaudio,
  title={TorchAudio: Building Blocks for Audio and Speech Processing},
  author={Yao-Yuan Yang and Moto Hira and Zhaoheng Ni and Anjali Chourdia and Artyom Astafurov and Caroline Chen and Ching-Feng Yeh and Christian Puhrsch and David Pollack and Dmitriy Genzel and Donny Greenberg and Edward Z. Yang and Jason Lian and Jay Mahadeokar and Jeff Hwang and Ji Chen and Peter Goldsborough and Prabhat Roy and Sean Narenthiran and Shinji Watanabe and Soumith Chintala and Vincent Quenneville-Bélair and Yangyang Shi},
  journal={arXiv preprint arXiv:2110.15018},
  year={2021}
}
```

Vincent QB's avatar
Vincent QB committed
60
61
62
63
64
65
Disclaimer on Datasets
----------------------

This is a utility library that downloads and prepares public datasets. We do not host or distribute these datasets, vouch for their quality or fairness, or claim that you have license to use the dataset. It is your responsibility to determine whether you have permission to use the dataset under the dataset's license.

If you're a dataset owner and wish to update any part of it (description, citation, etc.), or do not want your dataset to be included in this library, please get in touch through a GitHub issue. Thanks for your contribution to the ML community!
66
67
68
69
70
71

Pre-trained Model License
-------------------------

The pre-trained models provided in this library may have their own licenses or terms and conditions derived from the dataset used for training. It is your responsibility to determine whether you have permission to use the models for your use case.

72
For instance, SquimSubjective model is released under the Creative Commons Attribution Non Commercial 4.0 International (CC-BY-NC 4.0) license. See [the link](https://zenodo.org/record/4660670#.ZBtWPOxuerN) for additional details.
73
74

Other pre-trained models that have different license are noted in documentation. Please checkout the [documentation page](https://pytorch.org/audio/main/).