numpy
opencv-python
timm
torch
torchvision
torchsummary
tensorboardX
tqdm
einops
pandas
