torch>=1.4.0
torchvision
matplotlib
pillow
opencv-python
