opencv-python
numpy
matplotlib
torch==1.10.2
torchvision==0.11.3