timm==0.5.4
scikit-learn
opencv-python
mlflow
pycocotools
einops
tqdm