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