torch
torchvision
opencv-contrib-python
numpy
matplotlib