''' pip install tensorboardX imageio grad-cam sklearn matplotlib timm tqdm pretrainedmodels tensorboard '''