tensorboardX
matplotlib
einops
yaml
opencv-python
pandas
scikit-image
tqdm