matplotlib
opencv-python
torch
tensorboardX
tensorboard
torchvision
