0
0
Fork 0
mirror of https://github.com/boltgolt/howdy.git synced 2024-09-19 09:51:19 +02:00
howdy/learn.py
2018-01-05 01:59:44 +01:00

29 lines
857 B
Python

import subprocess
import time
import os
user = os.environ.get("USER")
if not os.path.exists("models"):
os.makedirs("models")
if not os.path.exists("models/" + user):
print("No face model folder found, creating one")
os.makedirs("models/" + user)
print("Learning face for the user account " + os.environ.get("USER"))
print("Please look straigt into the camera for 5 seconds")
time.sleep(2.5)
subprocess.call(["fswebcam", "-S", "30", "--no-banner", "-d", "/dev/video1", "./models/" + user + "/L.jpg"], stderr=open(os.devnull, "wb"))
time.sleep(.3)
subprocess.call(["fswebcam", "-S", "6", "--no-banner", "-d", "/dev/video1", "./models/" + user + "/M.jpg"], stderr=open(os.devnull, "wb"))
time.sleep(.3)
subprocess.call(["fswebcam", "--no-banner", "-d", "/dev/video1", "./models/" + user + "/S.jpg"], stderr=open(os.devnull, "wb"))
print("Done.")