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()
