From a2df523a433a94d2c8bb7d2b1bf82afbdefd4175 Mon Sep 17 00:00:00 2001 From: Sidharth Hulyalkar Date: Tue, 3 Oct 2023 21:50:52 -0500 Subject: [PATCH] update renamed variable file_set_id in FacemapModelTraining --- element_facemap/train_facial_model.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/element_facemap/train_facial_model.py b/element_facemap/train_facial_model.py index 4a40ebf..663b88c 100644 --- a/element_facemap/train_facial_model.py +++ b/element_facemap/train_facial_model.py @@ -293,12 +293,12 @@ def make(self, key): video_files = [find_full_path(fbe.get_facemap_root_data_dir(), fp).as_posix() for fp in (FacemapTrainFileSet.VideoFile & - {'video_set_id': key['video_set_id']}).fetch("file_path")] + {'file_set_id': key['file_set_id']}).fetch("file_path")] # manually specified .h5 keypoints file train_fileset = [find_full_path(fbe.get_facemap_root_data_dir(), fp).as_posix() for fp in (FacemapTrainFileSet.File & - {'file_set_id': key['video_set_id']}).fetch("file_path")] + {'file_set_id': key['file_set_id']}).fetch("file_path")] keypoints_file_name = (FacemapModelTrainingTask & key).fetch1("keypoints_filename")