bobby/open_dbm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add private module to create model object for movement api

Browse Source
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
Download Patch File Download Diff File Expand all files Collapse all files

52
opendbm/api_lib/movement/_facial_tremor.py Normal file
View File

@@ -0,0 +1,52 @@
import tempfile
from opendbm.api_lib.model import VideoModel
from opendbm.dbm_lib import fac_tremor_process
class FacialTremor(VideoModel):
def __init__(self):
super().__init__()
self._params = [
"fac_features_mean_5",
"fac_tremor_median_5",
"fac_disp_median_5",
"fac_corr_5",
"fac_features_mean_12",
"fac_tremor_median_12",
"fac_disp_median_12",
"fac_corr_12",
"fac_features_mean_8",
"fac_tremor_median_8",
"fac_disp_median_8",
"fac_corr_8",
"fac_features_mean_48",
"fac_tremor_median_48",
"fac_disp_median_48",
"fac_corr_48",
"fac_features_mean_54",
"fac_tremor_median_54",
"fac_disp_median_54",
"fac_corr_54",
"fac_features_mean_28",
"fac_tremor_median_28",
"fac_disp_median_28",
"fac_corr_28",
"fac_features_mean_51",
"fac_tremor_median_51",
"fac_disp_median_51",
"fac_corr_51",
"fac_features_mean_66",
"fac_tremor_median_66",
"fac_disp_median_66",
"fac_corr_66",
"fac_features_mean_57",
"fac_tremor_median_57",
"fac_disp_median_57",
"fac_corr_57",
]
def _fit_transform(self, path):
return fac_tremor_process(
path, f"{tempfile.gettempdir()}/", self.r_config, save=False
)