albumentations >= 1.4.5 timm datasets torchmetrics pycocotools