numpy>=1.17.2
torch>=1.8.1
packaging

[:python_version < "3.9"]
typing-extensions

[all]
pystoi<=0.3.3
pycocotools>2.0.0
torchvision>=0.8
lpips<=0.1.4
torch-fidelity<=0.3.0
scipy>1.0.0
transformers>=4.10.0
nltk>=3.6
regex>=2021.9.24
tqdm>=4.41.0

[audio]
pystoi<=0.3.3

[detection]
pycocotools>2.0.0
torchvision>=0.8

[image]
lpips<=0.1.4
torch-fidelity<=0.3.0
torchvision>=0.8
scipy>1.0.0

[multimodal]
transformers>=4.10.0

[test]
huggingface-hub<0.7
pytorch-msssim==0.2.1
coverage>5.2
types-setuptools
dython<=0.7.3
scikit-learn>1.0
netcal>1.0.0
types-tabulate
scipy>1.0.0
transformers>4.4.0
pypesq>1.2
phmdoctest>=1.1.1
requests<=2.28.2
types-PyYAML
fast-bss-eval>=0.1.0
bert_score==0.3.13
types-emoji
types-protobuf
fire<=0.5.0
pytest-doctestplus>=0.9.0
jiwer>=2.3.0
kornia>=0.6.7
mir-eval>=0.6
sacrebleu>=2.0.0
pytest-timeout<=2.1.0
pytest-cov>2.10
cloudpickle>1.3
pandas>1.0.0
torch_complex<=0.4.3
pytest-rerunfailures>=10.0
psutil<=5.9.4
pytest>=6.0.0
scikit-image>0.17.1
types-requests
rouge-score>0.1.0
types-six
mypy==0.982

[text]
nltk>=3.6
regex>=2021.9.24
tqdm>=4.41.0

[visual]
