30 lines
1.2 KiB
Python
30 lines
1.2 KiB
Python
import argparse
|
|
import face_recognition.comparison
|
|
import cv2
|
|
|
|
|
|
if __name__ == '__main__':
|
|
parser = argparse.ArgumentParser(description='Visualise face recognition algorithms.')
|
|
parser.add_argument('--camera', '-c', type=int, default=0,
|
|
help='Numeric id of the camera')
|
|
parser.add_argument('--fullscreen', '-f', action='store_true',
|
|
help='Display output full screen')
|
|
parser.add_argument('--clockwise', action='store_true',
|
|
help='Rotate clockwise')
|
|
parser.add_argument('--counter-clockwise', action='store_true',
|
|
help='Rotate counter clockwise')
|
|
parser.add_argument('--cascade', default='haarcascade_frontalface_alt2.xml',
|
|
help='Cascade XML file to use (opencv format)')
|
|
parser.add_argument('--output', default='saves',
|
|
help='Directory to store images (after pressing spacebar)')
|
|
|
|
args = parser.parse_args()
|
|
|
|
rotate = None
|
|
if args.clockwise:
|
|
rotate = cv2.ROTATE_90_CLOCKWISE
|
|
if args.counter_clockwise:
|
|
rotate = cv2.ROTATE_90_COUNTERCLOCKWISE
|
|
|
|
face_recognition.comparison.main(args.camera, rotate, args.fullscreen, args.cascade, args.output)
|