add private module to create model object for movement api
This commit is contained in:
11
opendbm/api_lib/movement/_eye_blink.py
Normal file
11
opendbm/api_lib/movement/_eye_blink.py
Normal 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)
|
||||
Reference in New Issue
Block a user