wf-pycocotools==2.0.1.4
torch
torchvision