b3ae6130ed4714852211a6e8cb56de66450da83c,scripts/extract.py,ExtractTrainingData,process,#ExtractTrainingData#,17

Before Change

        try:
            for filename in self.read_directory():
                image = cv2.imread(filename)
                for idx, face in self.get_faces(image):
                    resized_image = extractor.extract(image, face, 256)
                    output_file = self.output_dir / Path(filename).stem
                    cv2.imwrite(str(output_file) + str(idx) + Path(filename).suffix, resized_image)
                
        except Exception as e:
            print("Failed to extract from image: {}. Reason: {}".format(filename, e))

After Change

        extractor_name = "Align" // TODO Pass as argument
        self.extractor = PluginLoader.get_extractor(extractor_name)()
        self.faces_detected = 0
        processes = self.arguments.processes
        if processes != 1:
            files = list(self.read_directory())
            for _ in tqdm(pool_process(self.processFiles, files, processes=processes), total = len(files)):
                self.faces_detected +=1
        else:
            try:
                for filename in tqdm(self.read_directory()):
                    self.handleImage(filename)

            except Exception as e:
                print("Failed to extract from image: {}. Reason: {}".format(filename, e))

    def processFiles(self, filename):
        try:
            return self.handleImage(filename)
        except Exception as e:

Instances