X_val_aug, y_val_one_hot = augment_images_one_hot_labels(X_val, y_val, IMAGE_SIZE..., val_pipe, categories);