numpy 
opencv-python
torch
torchvision