einops
torch
torchvision
natsort
numpy
opencv-python-headless
yacs
scikit-image
scipy
tb-nightly