pytorch
torchvision
numpy
opencv-python
