53 lines
1.5 KiB
Python
53 lines
1.5 KiB
Python
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
|
|
)
|