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