torch
torchvision
matplotlib
opencv-python
pillow