Function test_2d_pose_estimation

tests/test_demo_human.py:34–58

Test that 2D pose estimation runs and produces output.

(test_image_path, test_output_dir)

def test_2d_pose_estimation(test_image_path, test_output_dir):
    """Test that 2D pose estimation runs and produces output."""
    # Import here to avoid import issues at collection time
    from fmpose3d.lib.hrnet.gen_kpts import gen_video_kpts as hrnet_pose
    from fmpose3d.lib.preprocess import h36m_coco_format, revise_kpts
    
    # Run 2D pose estimation
    keypoints, scores = hrnet_pose(test_image_path, det_dim=416, num_person=1, gen_output=True, type='image')
    
    # Check output shapes
    assert keypoints is not None, "Keypoints should not be None"
    assert scores is not None, "Scores should not be None"
    assert keypoints.shape[2] == 17, f"Expected 17 joints, got {keypoints.shape[2]}"
    assert keypoints.shape[3] == 2, f"Expected 2D coordinates, got shape {keypoints.shape[3]}"
    
    # Convert to H36M format
    keypoints, scores, valid_frames = h36m_coco_format(keypoints, scores)
    assert keypoints is not None, "H36M format conversion failed"
    
    # Save keypoints
    output_2d_dir = test_output_dir + 'input_2D/'
    os.makedirs(output_2d_dir, exist_ok=True)
    np.savez_compressed(output_2d_dir + 'keypoints.npz', reconstruction=keypoints)
    
    assert os.path.exists(output_2d_dir + 'keypoints.npz'), "Keypoints file should be created"


def test_demo_pipeline_runs(test_image_path):