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.")