torch
torchvision
opencv