def process(self): extractor_name = "Align" # TODO Pass as argument self.extractor = PluginLoader.get_extractor(extractor_name)() processes = self.arguments.processes try: if processes != 1: files = list(self.read_directory()) for filename, faces in tqdm(pool_process(self.processFiles, files, processes=processes), total=len(files)): self.num_faces_detected += 1 self.faces_detected[os.path.basename(filename)] = faces else: try: for filename in tqdm(self.read_directory()): image = cv2.imread(filename) self.faces_detected[os.path.basename( filename)] = self.handleImage(image, filename) except Exception as e: print( 'Failed to extract from image: {}. Reason: {}'.format( filename, e)) finally: self.write_alignments()
def extract_multi_process(self): """ Run the extraction on the correct number of processes """ for filename, faces in tqdm(pool_process( self.process_single_image, self.images.input_images, processes=self.args.processes), total=self.images.images_found): self.faces.num_faces_detected += 1 self.faces.faces_detected[os.path.basename(filename)] = faces
def extract_multi_process(self): """ Run the extraction on the correct number of processes """ for filename, faces in tqdm(pool_process(self.process_single_image, self.images.input_images, processes=self.args.processes), total=self.images.images_found, file=sys.stdout): self.faces.num_faces_detected += 1 self.faces.faces_detected[os.path.basename(filename)] = faces
def extract_multi_process(self): """ Run the extraction on the correct number of processes """ frame_no = 0 for filename, faces in tqdm(pool_process(self.process_single_image, self.images.input_images), total=self.images.images_found, file=sys.stdout): self.faces.num_faces_detected += 1 self.faces.faces_detected[os.path.basename(filename)] = faces frame_no += 1 if frame_no == self.save_interval: self.write_alignments() frame_no = 0
def extract_multi_process(self): """ Run the extraction on the correct number of processes """ frame_no = 0 for filename, faces in tqdm( pool_process( self.process_single_image, self.images.input_images), total=self.images.images_found, file=sys.stdout): self.faces.num_faces_detected += 1 self.faces.faces_detected[os.path.basename(filename)] = faces frame_no += 1 if frame_no == self.save_interval: self.write_alignments() frame_no = 0
def process(self): 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 process(self): extractor_name = "Align" # TODO Pass as argument self.extractor = PluginLoader.get_extractor(extractor_name)() processes = self.arguments.processes try: if processes != 1: files = list(self.read_directory()) for filename, faces in tqdm(pool_process(self.processFiles, files, processes=processes), total = len(files)): self.num_faces_detected += 1 self.faces_detected[os.path.basename(filename)] = faces else: try: for filename in tqdm(self.read_directory()): image = cv2.imread(filename) self.faces_detected[os.path.basename(filename)] = self.handleImage(image, filename) except Exception as e: print('Failed to extract from image: {}. Reason: {}'.format(filename, e)) finally: self.write_alignments()