def pre_process_data(images): """Pre-process the data folder to find all images and crop all faces. Return: ImageSet of all faces. """ face_haar_cascade = HaarCascade('face2.xml') cropped_images = ImageSet() for image in images: cropped_images.extend(get_faces(image, face_haar_cascade)) return cropped_images