From c55af8290d9e96fd727d0ca06c8e6753ad4aa855 Mon Sep 17 00:00:00 2001 From: Guillaume HEUSCH <guillaume.heusch@idiap.ch> Date: Fri, 8 Feb 2019 15:37:20 +0100 Subject: [PATCH] [script] python3 compat for extracting eyes center --- bob/db/fargo/scripts/extract_eyes_center.py | 20 +++++++------------- 1 file changed, 7 insertions(+), 13 deletions(-) diff --git a/bob/db/fargo/scripts/extract_eyes_center.py b/bob/db/fargo/scripts/extract_eyes_center.py index e145a00..b957c2e 100644 --- a/bob/db/fargo/scripts/extract_eyes_center.py +++ b/bob/db/fargo/scripts/extract_eyes_center.py @@ -146,15 +146,6 @@ def get_eyes_center(landmarks): reye_y = int(0.5 * (landmarks['1'][0] + landmarks['2'][0])) leye_x = int(0.5 * (landmarks['3'][1] + landmarks['4'][1])) leye_y = int(0.5 * (landmarks['3'][0] + landmarks['4'][0])) - - # print "============================" - # print "right eye right corner (y,x) = {0}".format(landmarks['1']) - # print "right eye left corner (y,x) = {0}".format(landmarks['2']) - # print "right eye center (y,x) = {0}".format((reye_y, reye_x)) - # print "----------------------------" - # print "left eye right corner (y,x) = {0}".format(landmarks['4']) - # print "left eye right corner (y,x) = {0}".format(landmarks['3']) - # print "left eye center (y,x) = {0}".format((leye_y, leye_x)) return (reye_x, reye_y, leye_x, leye_y) @@ -516,7 +507,7 @@ def annotate(image_file, eyes, annotation_dir): f.close() os.system('./bin/annotate.py ' + image_file + ' 00.pos --output temp.txt') - raw_input("Press Enter to terminate.") + input("Press Enter to terminate.") # read eyes center from temp annotation file f1 = open('temp.txt', 'r') @@ -567,7 +558,10 @@ def main(user_input=None): os.mkdir(args['--eyesdir']) logfile = open(args['--log'], 'w') - + + # python 2 / 3 compatibility + from builtins import input + # to compute various stats total_counter = 0 inexisting_counter = 0 @@ -662,7 +656,7 @@ def main(user_input=None): pyplot.show() # ask for re-annotation (i.e. reprojection is not good) - reannotate = raw_input("Want to re-annotate color ? [y/n]: ") + reannotate = input("Want to re-annotate color ? [y/n]: ") if reannotate == 'y': # get eyes position in the color frame @@ -696,7 +690,7 @@ def main(user_input=None): pyplot.show() # ask for re-annotation (i.e. reprojection is not good) - reannotate = raw_input("Want to re-annotate IR ? [y/n]: ") + reannotate = input("Want to re-annotate IR ? [y/n]: ") if reannotate == 'y': # get eyes position in the IR frame -- GitLab