torch
torchvision
torchaudio
pycocotools
Pillow
numpy