numpy
torch
torchvision
Pillow
opencv-python
matplotlib
