r
/
face_recognition
1
0
Fork 0
Code Issues Pull Requests Releases 2 Activity

Fix image saving for windows + error message on failure

This commit is contained in:
Ruben van de Ven 2020-10-02 17:32:30 +02:00
parent 2af58fc170
commit 9d1b523c56
1 changed files with 11 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
face_recognition/comparison.py
View File

@ -606,11 +606,19 @@ def display(image_res, q1, q2, q3, q4, fullscreen, output_dir):
logger.info("Show frame until %f", override_until)
# save images:
name = datetime.datetime.now().isoformat(timespec='seconds')
cv2.imwrite(os.path.join(output_dir, f'{name}.png'),override_image)
name = datetime.datetime.now().isoformat(timespec='seconds').replace(':','-')
filename = os.path.join(output_dir, f'{name}.png')
print(f"Save to {filename}")
r=cv2.imwrite(filename, override_image)
if not r:
raise RuntimeError(f"Could not save image {filename}")
for result in results:
result_img =result.draw_detections(include_title = True)
cv2.imwrite(os.path.join(output_dir, f'{name}-{result.algorithm}.png'), result_img)
filename = os.path.join(output_dir, f'{name}-{result.algorithm}.png')
r = cv2.imwrite(filename, result_img)
if not r:
raise RuntimeError(f"Could not save image {filename}")