25 lines
638 B
Python
25 lines
638 B
Python
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
|
|
)
|