torch
einops
numpy
opencv-python
matplotlib
yacs