torch
numpy
argparse
sklearn
torchvision
kornia