diff --git a/element_deeplabcut/model.py b/element_deeplabcut/model.py index 7060aa5..1462f9b 100644 --- a/element_deeplabcut/model.py +++ b/element_deeplabcut/model.py @@ -598,7 +598,7 @@ def make(self, key): get_dlc_root_data_dir(), dlc_model["project_path"] ) - # Triger PoseEstimation, + # Triger PoseEstimation if task_mode == "trigger": dlc_reader.do_pose_estimation( video_filepaths, diff --git a/element_deeplabcut/readers/dlc_reader.py b/element_deeplabcut/readers/dlc_reader.py index faef51c..30b7973 100644 --- a/element_deeplabcut/readers/dlc_reader.py +++ b/element_deeplabcut/readers/dlc_reader.py @@ -154,7 +154,7 @@ def do_pose_estimation( dlc_model, project_path, output_dir, - videotype=None, + videotype="", gputouse=None, save_as_csv=False, batchsize=None, diff --git a/requirements.txt b/requirements.txt index d04ca5b..fdf7d7c 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,2 +1,2 @@ datajoint>=0.13 -deeplabcut \ No newline at end of file +deeplabcut>=2.2.1.1 \ No newline at end of file