File size: 309 Bytes
cc396fd
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
"""Online current estimator."""

import numpy as np

from experiments.shared.src.vision.pose_from_image import estimate_pose_from_clean_image

def estimate_current(history, config):
    poses = np.stack([estimate_pose_from_clean_image(img) for img in history], axis=0)
    return poses[-1, :2] - poses[0, :2]