opencv-python
torch
torchvision
scipy