torch
torchvision
numpy
opencv-python
h5py
tqdm
einops
pathlib
