torch
torchmetrics
pycocotools
