torch
transforms
torchvision
opencv-python>=4.1.1