imt-ring 1.6.2__py3-none-any.whl → 1.6.3__py3-none-any.whl

{imt_ring-1.6.2.dist-info → imt_ring-1.6.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: imt-ring
-Version: 1.6.2
+Version: 1.6.3
 Summary: RING: Recurrent Inertial Graph-based Estimator
 Author-email: Simon Bachhuber <simon.bachhuber@fau.de>
 Project-URL: Homepage, https://github.com/SimiPixel/ring

{imt_ring-1.6.2.dist-info → imt_ring-1.6.3.dist-info}/RECORD

@@ -8,15 +8,15 @@ ring/algorithms/_random.py,sha256=fc26yEQjSjtf0NluZ41CyeGIRci0ldrRlThueHR9H7U,14
 ring/algorithms/dynamics.py,sha256=_TwclBXe6vi5C5iJWAIeUIJEIMHQ_1QTmnHvCEpVO0M,10867
 ring/algorithms/jcalc.py,sha256=bM8VARgqEiVPy7632geKYGk4MZddZfI8XHdW5kXF3HI,28594
 ring/algorithms/kinematics.py,sha256=DOboHI517Vx0pRJUFZtZPmK_qFaiKiQe-37B-M0aC-c,7422
-ring/algorithms/sensors.py,sha256=QSIcU_sEB_tRo-ADD_66ZD01LMmJlSG0op6YnM-Gai8,17965
+ring/algorithms/sensors.py,sha256=06x7RfhoQ6dx1B_TAEuCKxNTiicQDDBxcmzRtsCAxsM,18125
 ring/algorithms/custom_joints/__init__.py,sha256=fzeE7TdUhmGgbbFAyis1tKcyQ4Fo8LigDwD3hUVnH_w,316
 ring/algorithms/custom_joints/rr_imp_joint.py,sha256=_YJK0p8_0MHFtr1NuGnNZoxTbwaMQyUjYv7EtsPiU3A,2402
 ring/algorithms/custom_joints/rr_joint.py,sha256=jnRtjtOCALMaq2_0bcu2d7qgfQ6etXpoh43MioRaDmY,1000
 ring/algorithms/custom_joints/suntay.py,sha256=tOEGM304XciHO4pmvxr4faA4xXVO4N2HlPdFmXKbcrw,16726
 ring/algorithms/generator/__init__.py,sha256=bF-CW3x2x-o6KWESKy-DuxzZPh3UNSjJb_MaAcSHGsQ,277
-ring/algorithms/generator/base.py,sha256=KQSg9uhhR-rC563busVFx4gJrqOx3BXdaChozO9gwTA,14224
+ring/algorithms/generator/base.py,sha256=ZW6tJSUiNKtRCYIsbCf6e6kRBbmuCLzbZj4ppNLVwJY,14368
 ring/algorithms/generator/batch.py,sha256=ylootnXmj-JyuB_f5OCknHst9wFKO3gkjQbMrFNXY2g,2513
-ring/algorithms/generator/finalize_fns.py,sha256=L_5wIVA7g0P4P2U6EmgcvsoI-YuF3TOaHBwk5_oEaUU,9077
+ring/algorithms/generator/finalize_fns.py,sha256=LUw1Wc2YrmMRRh4RF704ob3bZOXktAZAbbLoBm_p1yw,9131
 ring/algorithms/generator/motion_artifacts.py,sha256=2VJbldVDbI3PSyboshIbtYvSAKzBBwGV7cQfYjqvluM,9167
 ring/algorithms/generator/pd_control.py,sha256=XJ_Gd5AkIRh-jBrMfQyMXjVwhx2gCNHznjzFbmAwhZs,5767
 ring/algorithms/generator/setup_fns.py,sha256=MFz3czHBeWs1Zk1A8O02CyQpQ-NCyW9PMpbqmKit6es,1455
@@ -83,7 +83,7 @@ ring/utils/randomize_sys.py,sha256=G_vBIo0OwQkXL2u0djwbaoaeb02C4LQCTNNloOYIU2M,3
 ring/utils/utils.py,sha256=k7t-QxMWrNRnjfNB9rSobmLCmhJigE8__gkT-Il0Ee4,6492
 ring/utils/register_gym_envs/__init__.py,sha256=PtPIRBQJ16339xZ9G9VpvqrvcGbQ_Pk_SUz4tQPa9nQ,94
 ring/utils/register_gym_envs/saddle.py,sha256=tA5CyW_akSXyDm0xJ83CtOrUMVElH0f9vZtEDDJQalI,4422
-imt_ring-1.6.2.dist-info/METADATA,sha256=S_LlVrmdRPQCzT5aeRoSWyOQ3eBJBL1D33tUgXMUEso,3104
-imt_ring-1.6.2.dist-info/WHEEL,sha256=Wyh-_nZ0DJYolHNn1_hMa4lM7uDedD_RGVwbmTjyItk,91
-imt_ring-1.6.2.dist-info/top_level.txt,sha256=EiT790-lAyi8iwTzJArH3f2k77rwhDn00q-4PlmvDQo,5
-imt_ring-1.6.2.dist-info/RECORD,,
+imt_ring-1.6.3.dist-info/METADATA,sha256=uFTNWR0YQbQLE50Oby-D2H3NSpdycgCwzKaz_UxxSP8,3104
+imt_ring-1.6.3.dist-info/WHEEL,sha256=Wyh-_nZ0DJYolHNn1_hMa4lM7uDedD_RGVwbmTjyItk,91
+imt_ring-1.6.3.dist-info/top_level.txt,sha256=EiT790-lAyi8iwTzJArH3f2k77rwhDn00q-4PlmvDQo,5
+imt_ring-1.6.3.dist-info/RECORD,,

ring/algorithms/generator/base.py

@@ -30,6 +30,7 @@ class RCMG:
         add_X_jointaxes_kwargs: dict = dict(),
         add_y_relpose: bool = False,
         add_y_rootincl: bool = False,
+        add_y_rootincl_kwargs: dict = dict(),
         sys_ml: Optional[base.System] = None,
         randomize_positions: bool = False,
         randomize_motion_artifacts: bool = False,
@@ -67,6 +68,7 @@ class RCMG:
                     add_X_jointaxes_kwargs=add_X_jointaxes_kwargs,
                     add_y_relpose=add_y_relpose,
                     add_y_rootincl=add_y_rootincl,
+                    add_y_rootincl_kwargs=add_y_rootincl_kwargs,
                     sys_ml=sys_ml,
                     randomize_positions=randomize_positions,
                     randomize_motion_artifacts=randomize_motion_artifacts,
@@ -232,6 +234,7 @@ def _build_mconfig_batched_generator(
     add_X_jointaxes_kwargs: dict,
     add_y_relpose: bool,
     add_y_rootincl: bool,
+    add_y_rootincl_kwargs: dict,
     sys_ml: base.System,
     randomize_positions: bool,
     randomize_motion_artifacts: bool,

ring/algorithms/generator/finalize_fns.py

@@ -77,12 +77,13 @@ class RelPose:
 
 
 class RootIncl:
-    def __init__(self, sys: base.System):
+    def __init__(self, sys: base.System, **kwargs):
         self.sys = sys
+        self.kwargs = kwargs
 
     def __call__(self, Xy, extras):
         (X, y), (key, q, x, sys_x) = Xy, extras
-        y_root_incl = sensors.root_incl(self.sys, x, sys_x)
+        y_root_incl = sensors.root_incl(self.sys, x, sys_x, **self.kwargs)
         y = utils.dict_union(y, y_root_incl)
         return (X, y), (key, q, x, sys_x)
 

ring/algorithms/sensors.py

@@ -330,7 +330,10 @@ def rel_pose(
 
 
 def root_incl(
-    sys: base.System, x: base.Transform, sys_x: base.System
+    sys: base.System,
+    x: base.Transform,
+    sys_x: base.System,
+    child_to_parent: bool = False,
 ) -> dict[str, jax.Array]:
     # (time, nlinks, 4) -> (nlinks, time, 4)
     rots = x.rot.transpose((1, 0, 2))
@@ -341,8 +344,10 @@ def root_incl(
     def f(_, __, name: str, parent: int):
         if parent != -1:
             return
-        q_eps_to_i = maths.quat_project(rots[l_map[name]], jnp.array([0.0, 0, 1]))[1]
-        y[name] = maths.quat_inv(q_eps_to_i)
+        q_i = maths.quat_project(rots[l_map[name]], jnp.array([0.0, 0, 1]))[1]
+        if child_to_parent:
+            q_i = maths.quat_inv(q_i)
+        y[name] = q_i
 
     sys.scan(f, "ll", sys.link_names, sys.link_parents)
 
@@ -350,7 +355,10 @@ def root_incl(
 
 
 def root_full(
-    sys: base.System, x: base.Transform, sys_x: base.System
+    sys: base.System,
+    x: base.Transform,
+    sys_x: base.System,
+    child_to_parent: bool = False,
 ) -> dict[str, jax.Array]:
     # (time, nlinks, 4) -> (nlinks, time, 4)
     rots = x.rot.transpose((1, 0, 2))
@@ -361,8 +369,10 @@ def root_full(
     def f(_, __, name: str, parent: int):
         if parent != -1:
             return
-        q_eps_to_i = rots[l_map[name]]
-        y[name] = maths.quat_inv(q_eps_to_i)
+        q_i = rots[l_map[name]]
+        if child_to_parent:
+            q_i = maths.quat_inv(q_i)
+        y[name] = q_i
 
     sys.scan(f, "ll", sys.link_names, sys.link_parents)