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,19 @@
import tempfile
from opendbm.api_lib.model import VideoModel
from opendbm.dbm_lib import run_head_movement
class HeadMovement(VideoModel):
def __init__(self):
super().__init__()
self._params = [
"mov_headvel",
"mov_hposepitch",
"mov_hposeyaw",
"mov_hposeroll",
"mov_hposedist",
]
def _fit_transform(self, path):
return run_head_movement(path, f"{tempfile.gettempdir()}/", self.r_config)