torch>=1.6
torchvision
cv2
