add private module to create model object for movement api
This commit is contained in:
24
opendbm/api_lib/movement/_eye_gaze.py
Normal file
24
opendbm/api_lib/movement/_eye_gaze.py
Normal file
@@ -0,0 +1,24 @@
|
||||
import tempfile
|
||||
|
||||
from opendbm.api_lib.model import VideoModel
|
||||
from opendbm.dbm_lib.dbm_features.raw_features.movement.eye_gaze import run_eye_gaze
|
||||
|
||||
|
||||
class EyeGaze(VideoModel):
|
||||
def __init__(self):
|
||||
super().__init__()
|
||||
self._params = [
|
||||
"mov_lefteyex",
|
||||
"mov_lefteyey",
|
||||
"mov_lefteyez",
|
||||
"mov_righteyex",
|
||||
"mov_righteyey",
|
||||
"mov_righteyez",
|
||||
"mov_leyedisp",
|
||||
"mov_reyedisp",
|
||||
]
|
||||
|
||||
def _fit_transform(self, path):
|
||||
return run_eye_gaze(
|
||||
path, f"{tempfile.gettempdir()}/", self.r_config, save=False
|
||||
)
|
||||
Reference in New Issue
Block a user