pytest
tqdm
opencv-python-headless
numpy
ptflops
torch
torchvision
grad-cam
pillow
