pre-commit.ci status

Split the image classification dataset into train and val. Add data augmentation on training data