glob os time PIL albumentations cv2 numpy matplotlib pandas scipy segmentation_models_pytorch sklearn torch torchvision tqdm tensorflow