12 lines
436 B
Python
12 lines
436 B
Python
from opendbm.api_lib.model import DLIB_SHAPE_MODEL, VideoModel
|
|
from opendbm.dbm_lib.dbm_features.raw_features.movement.eye_blink import run_eye_blink
|
|
|
|
|
|
class EyeBlink(VideoModel):
|
|
def __init__(self):
|
|
super().__init__()
|
|
self._params = ["mov_blink_ear", "mov_blinkframes", "mov_blinkdur", "fps"]
|
|
|
|
def _fit_transform(self, path):
|
|
return run_eye_blink(path, ".", self.r_config, DLIB_SHAPE_MODEL, save=False)
|