torch
torchvision
Pillow
colorlog
numpy
psutil
scikit-image
kornia
neptune-client