torch
numpy
scikit-image
omegaconf
easydict
tqdm
einops
