argparse
opencv-python
imageio
python-math
matplotlib
numpy
Pillow
torch
torchvision