X_train_aug, y_train_one_hot = augment_images_one_hot_labels(X_train, y_train, IMAGE_SIZE..., train_pipe, categories);