add private module to create model object for movement api

This commit is contained in:
jordi.hasianta
2022-09-15 19:46:33 +07:00
parent ded93afb86
commit 760c9617be
6 changed files with 125 additions and 0 deletions

View File

@@ -0,0 +1,11 @@
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)