Skip to content

Commit

Permalink
Fix pre-commit checks
Browse files Browse the repository at this point in the history
  • Loading branch information
KavitShah1998 committed Dec 21, 2023
1 parent c242899 commit b1f8522
Show file tree
Hide file tree
Showing 2 changed files with 7 additions and 3 deletions.
4 changes: 3 additions & 1 deletion aria_data_loaders/aria_data_utils/aria_live_streamer.py
Original file line number Diff line number Diff line change
Expand Up @@ -570,7 +570,9 @@ def main(do_update_iptables: bool, read_only: bool, debug: bool):
outputs = {}
aria_object = AriaLiveReader(verbose=True)
outputs = aria_object.initialize_april_tag_detector(outputs=outputs)
outputs = aria_object.initialize_object_detector(outputs=outputs, object_labels=[object_name])
outputs = aria_object.initialize_object_detector(
outputs=outputs, object_labels=[object_name]
)
aria_object.connect()
while not rospy.is_shutdown():
frame = aria_object.get_latest_frame()
Expand Down
6 changes: 4 additions & 2 deletions aria_data_loaders/aria_data_utils/math_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,12 +2,13 @@

import numpy as np
import sophus as sp

# from bosdyn.api.geometry_pb2 import SE3Pose
# from geometry_msgs.msg import Pose, PoseStamped, TransformStamped
from scipy.spatial.transform import Rotation

FILTER_DIST = 2.4 # in meters (distance for valid detection)
FIXED_LIST_LENGTH = 10 # Fixed length of the a_T_b_{position,orientation} lists used for average computation
FIXED_LIST_LENGTH = 10 # Fixed length of the a_T_b_{position,orientation} lists used for average computation


def compute_avg_sophus_SE3_from_nplist(
Expand Down Expand Up @@ -41,6 +42,7 @@ def compute_avg_sophus_SE3_from_nplist(

return avg_sp_se3


def get_running_avg_world_T_b(
current_avg_world_T_b: sp.SE3,
world_T_b_position_list: List[np.ndarray],
Expand Down Expand Up @@ -106,4 +108,4 @@ def get_running_avg_world_T_b(
# If the latest detection is not within the filter distance, then do not update the list
avg_world_T_b = current_avg_world_T_b

return world_T_b_position_list, world_T_b_quaternion_list,avg_world_T_b
return world_T_b_position_list, world_T_b_quaternion_list, avg_world_T_b

0 comments on commit b1f8522

Please sign in to comment.