timm==0.5.4
scikit-image
ptflops
easydict
PyYAML
pillow
opencv-python
tqdm
pyyaml
einops