PyYAML
argparse
matplotlib
numpy
opencv-python
scipy
tensorboardX
torchvision
torch