numpy
pandas
torch
torchvision
torchmetrics>=0.11.0
pycocotools
scikit-learn
