opencv-python
matplotlib
ultralytics
torch
albumentations
pycocotools
lightning
timm
scipy
torchvision
gdown
jupyter
