OpenSeeFace/feature.py
Cowwan 526c037336 Made FeatureExtractor a separate file
added documentation to featureExtractor.py to help people customize it
2024-02-06 08:16:39 -05:00

56 lines
2 KiB
Python

import math
#got tired of trying to remember which order I had to do the nested min and max
def clamp (value, minimum, maxium):
return max(min(value,maxium),minimum)
def rotate(origin, point, a):
x, y = point - origin
cosa = math.cos(-a)
sina = math.sin(-a)
qx = origin[0] + cosa * x - sina * y
qy = origin[1] + sina * x + cosa * y
return qx, qy
#I redid a lot of this so it worked well for me
#idk if it'll work well for other people
class Feature():
def __init__(self, alpha=0.1, decay=0.0005, curve=1, scaleType = 1):
self.min = None
self.max = None
self.alpha = alpha
self.decay = decay
self.last = 0.0
self.curve = curve
self.scaleType = scaleType
def update(self, x):
new = self.update_state(x)
self.last = self.last * self.alpha + new * (1 - self.alpha)
return self.last
def update_state(self, x):
if self.min is None or self.max is None:
self.min = x - 0.00001
self.max = x + 0.00001
self.min = min(self.min, x)
self.max = max(self.max, x)
center = (self.min+self.max)/2
#The Min and Max decay slightly toward each other every frame allowing the range to change dynamically
self.min = (self.min * (1 - self.decay)) + (self.max * self.decay)
self.max = (self.max * (1 - self.decay)) + (self.min * self.decay)
if self.scaleType == 1:
#Returns a value between -1 and 1 in relation to the maximum range
if x < center:
return -pow(clamp((x - center) / (self.min - center), 0, 1), self.curve)
elif x > center:
return pow(clamp((x - center) / (self.max - center), 0, 1), self.curve)
return 0
if self.scaleType == 2:
#Returns a value between 0 and 1 in relation to the maximum range
return pow(clamp((x - self.min) / (self.max - self.min), 0, 1), self.curve)
return 0