debug causal

open-mmlab · May 30, 2023 · 9d689ad · 9d689ad
1 parent 94f636a
commit 9d689ad
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/demo/body3d_pose_lifter_demo.py b/demo/body3d_pose_lifter_demo.py
@@ -386,7 +386,7 @@ def main():
         pose_results_2d = extract_pose_sequence(
             pose_est_results_list,
             frame_idx=frame_idx,
-            causal=pose_lifter.causal,
+            causal=pose_lift_dataset.get('causal', False),
             seq_len=pose_lift_dataset.get('seq_len', 1),
             step=pose_lift_dataset.get('seq_step', 1))
 

diff --git a/mmpose/apis/inference_3d.py b/mmpose/apis/inference_3d.py
@@ -180,7 +180,7 @@ def inference_pose_lifter_model(model,
     init_default_scope(model.cfg.get('default_scope', 'mmpose'))
     pipeline = Compose(model.cfg.test_dataloader.dataset.pipeline)
 
-    causal = model.causal
+    causal = model.cfg.test_dataloader.dataset.get('causal', False)
     target_idx = -1 if causal else len(pose_results_2d) // 2
 
     dataset_info = model.dataset_meta