einops
mlflow
opencv-python
pytorch-lightning==1.1.4
torch==1.7.1
torchvision==0.8.2
tqdm
yacs