176 lines
5.0 KiB
Python
176 lines
5.0 KiB
Python
#!/usr/bin/python3
|
|
# -*- coding: utf-8 -*-
|
|
# -*- mode: Python -*-
|
|
|
|
'''
|
|
|
|
A Face tracker
|
|
v0.1.0
|
|
|
|
Get all points fom redis /trckr/frame/WSclientID points
|
|
|
|
Licensed under GNU GPLv3
|
|
|
|
by cocoa and Sam Neurohack
|
|
|
|
'''
|
|
|
|
from __future__ import print_function
|
|
import time
|
|
import argparse
|
|
import sys
|
|
import redis
|
|
import ast
|
|
|
|
name="generator::trckr"
|
|
|
|
|
|
def debug(*args, **kwargs):
|
|
if( verbose == False ):
|
|
return
|
|
print(*args, file=sys.stderr, **kwargs)
|
|
|
|
|
|
argsparser = argparse.ArgumentParser(description="Face tracking generator")
|
|
argsparser.add_argument("-f","--fps",help="Frame Per Second",default=30,type=int)
|
|
argsparser.add_argument("-v","--verbose",action="store_true",help="Verbose output")
|
|
argsparser.add_argument("-i","--id",help="Trckr client ID",default="0",type=str)
|
|
argsparser.add_argument("-s","--server",help="redis server IP (127.0.0.1 by default)", type=str)
|
|
args = argsparser.parse_args()
|
|
|
|
fps=args.fps
|
|
verbose=args.verbose
|
|
idclient = args.id
|
|
|
|
if args.server:
|
|
redisIP = args.server
|
|
else:
|
|
redisIP = "127.0.0.1"
|
|
|
|
|
|
|
|
optimal_looptime = 1 / fps
|
|
debug(name+" optimal looptime "+str(optimal_looptime))
|
|
|
|
color = 65280
|
|
|
|
|
|
def rgb2int(rgb):
|
|
return int('0x%02x%02x%02x' % tuple(rgb),0)
|
|
|
|
# Useful variables init.
|
|
white = rgb2int((255,255,255))
|
|
red = rgb2int((255,0,0))
|
|
blue = rgb2int((0,0,255))
|
|
green = rgb2int((0,255,0))
|
|
|
|
#
|
|
# Redis functions
|
|
#
|
|
|
|
r = redis.StrictRedis(host=redisIP , port=6379, db=0)
|
|
|
|
# read from redis key
|
|
def fromKey(keyname):
|
|
|
|
return r.get(keyname)
|
|
|
|
# Write to redis key
|
|
def toKey(keyname,keyvalue):
|
|
return r.set(keyname,keyvalue)
|
|
|
|
#
|
|
# Trckr faces
|
|
#
|
|
|
|
TrckrPts = [[159.39, 137.68], [155.12, 159.31], [155.56, 180.13], [159.81, 201.6], [170.48, 220.51], [187.46, 234.81], [208.4, 244.68], [229.46, 248.21], [246.44, 244.91], [259.69, 234.83], [270.95, 221.51], [278.54, 204.66], [283.53, 185.63], [286.27, 165.79], [284.72, 144.84], [280.06, 125.01], [274.35, 118.7], [260.71, 117.23], [249.52, 118.86], [182.04, 121.5], [193.63, 114.79], [210.24, 114.77], [222.35, 117.57], [190.6, 137.49], [203.59, 132.42], [214.75, 137.58], [203.04, 140.46], [203.32, 136.53], [272.45, 141.57], [263.33, 135.42], [250.31, 138.89], [262.15, 143.27], [261.99, 139.37], [235.82, 131.74], [221.87, 156.09], [213.66, 165.88], [219.28, 173.53], [236.3, 175.25], [249.02, 174.4], [254.22, 167.81], [248.83, 157.39], [237.94, 147.51], [227.01, 168.39], [245.68, 170.02], [204.94, 197.32], [217.56, 192.77], [228.27, 190.55], [234.66, 192.19], [240.47, 191.09], [247.96, 193.87], [254.52, 199.19], [249.35, 204.25], [242.74, 207.16], [233.2, 207.87], [222.13, 206.52], [212.44, 203.09], [220.34, 198.74], [233.31, 200.04], [244.0, 199.6], [244.27, 197.8], [233.81, 197.44], [220.88, 196.99], [239.57, 162.69], [196.52, 133.86], [210.2, 133.98], [209.43, 139.41], [196.59, 139.47], [268.99, 137.59], [256.36, 136.02], [255.95, 141.5], [267.9, 142.85]]
|
|
toKey('/trckr/frame/0',str(TrckrPts))
|
|
|
|
# get absolute face position points
|
|
def getPART(TrckrPts, pose_points):
|
|
|
|
dots = []
|
|
#debug(pose_points)
|
|
#debug(TrckrPts)
|
|
for dot in pose_points:
|
|
dots.append((TrckrPts[dot][0], TrckrPts[dot][1],0))
|
|
#debug(dots)
|
|
return dots
|
|
|
|
|
|
# Face keypoints
|
|
def face(TrckrPts):
|
|
pose_points = [0,1,2,3,4,5,6,7,8,9,10,11,12,13,14]
|
|
return getPART(TrckrPts, pose_points)
|
|
|
|
def browL(TrckrPts):
|
|
pose_points = [15,16,17,18]
|
|
return getPART(TrckrPts, pose_points)
|
|
|
|
def browR(TrckrPts):
|
|
pose_points = [22,21,20,19]
|
|
return getPART(TrckrPts, pose_points)
|
|
|
|
def eyeR(TrckrPts):
|
|
pose_points = [25,64,24,63,23,66,26,65,25]
|
|
return getPART(TrckrPts, pose_points)
|
|
|
|
def eyeL(TrckrPts):
|
|
pose_points = [28,67,29,68,30,69,31,28]
|
|
return getPART(TrckrPts, pose_points)
|
|
|
|
def pupR(TrckrPts):
|
|
pose_points = [27]
|
|
return getPART(TrckrPts, pose_points)
|
|
|
|
def pupL(TrckrPts):
|
|
pose_points = [32]
|
|
return getPART(TrckrPts, pose_points)
|
|
|
|
|
|
def nose1(TrckrPts):
|
|
pose_points = [62,41,33]
|
|
return getPART(TrckrPts, pose_points)
|
|
|
|
def nose2(TrckrPts):
|
|
pose_points = [40,39,38,43,37,42,36,35,34]
|
|
return getPART(TrckrPts, pose_points)
|
|
|
|
def mouth(TrckrPts):
|
|
pose_points = [50,49,48,47,46,45,44,55,54,53,52,51,50]
|
|
return getPART(TrckrPts, pose_points)
|
|
|
|
def mouthfull(TrckrPts):
|
|
pose_points = [50,49,48,47,46,45,44,55,54,53,52,51,50,59,60,61,44,56,57,58,50]
|
|
return getPART(TrckrPts, pose_points)
|
|
|
|
|
|
while True:
|
|
|
|
start = time.time()
|
|
shape =[]
|
|
points = ast.literal_eval(fromKey('/trckr/frame/'+idclient).decode('ascii'))
|
|
shape.append(browL(points))
|
|
shape.append(eyeL(points))
|
|
shape.append(browR(points))
|
|
shape.append(eyeR(points))
|
|
shape.append(pupL(points))
|
|
shape.append(pupR(points))
|
|
shape.append(nose1(points))
|
|
shape.append(nose2(points))
|
|
shape.append(mouthfull(points))
|
|
line = str(shape)
|
|
line = line.replace("(",'[')
|
|
line = line.replace(")",']')
|
|
line = "[{}]".format(line)
|
|
print(line, flush=True);
|
|
#debug(shape)
|
|
#print(shape, flush=True);
|
|
|
|
looptime = time.time() - start
|
|
if( looptime < optimal_looptime ):
|
|
time.sleep( optimal_looptime - looptime)
|
|
debug(name+" micro sleep:"+str( optimal_looptime - looptime))
|
|
|
|
|