numpy
torch
pycocotools
