torch
ultralytics
albumentations
scikit-learn
matplotlib