torch

torchvision

opencv-python

thop

scipy

matplotlib

numpy

imageio

pycocotools

albumentations