The repo requires cv2 (conda install -c menpo opencv), tensorboardX >= 1.7 (pip install tensorboardX), joblib, tqdm, ipdb, numpy, matplotlib, wandb, pytorch (1.7), torchvision.