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