torch==1.11.0
torchvision==0.12.0
pycocotools
opencv-python
wandb
imageio
scipy