Code update

PiperOrigin-RevId: 662918283
google-research · Aug 14, 2024 · 0687e8e · 0687e8e
1 parent b622f8c
commit 0687e8e
Show file tree

Hide file tree

Showing 43 changed files with 49 additions and 44 deletions.
diff --git a/human_scene_transformer/__init__.py b/human_scene_transformer/__init__.py
@@ -1,4 +1,4 @@
-# Copyright 2023 The human_scene_transformer Authors.
+# Copyright 2024 The human_scene_transformer Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/human_scene_transformer/data/box_utils.py b/human_scene_transformer/data/box_utils.py
@@ -1,4 +1,4 @@
-# Copyright 2023 The human_scene_transformer Authors.
+# Copyright 2024 The human_scene_transformer Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/human_scene_transformer/data/extract_robot_odometry_from_rosbag.py b/human_scene_transformer/data/extract_robot_odometry_from_rosbag.py
@@ -1,4 +1,4 @@
-# Copyright 2023 The human_scene_transformer Authors.
+# Copyright 2024 The human_scene_transformer Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/human_scene_transformer/data/jrdb_preprocess_test.py b/human_scene_transformer/data/jrdb_preprocess_test.py
@@ -1,4 +1,4 @@
-# Copyright 2023 The human_scene_transformer Authors.
+# Copyright 2024 The human_scene_transformer Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/human_scene_transformer/data/jrdb_preprocess_train.py b/human_scene_transformer/data/jrdb_preprocess_train.py
@@ -1,4 +1,4 @@
-# Copyright 2023 The human_scene_transformer Authors.
+# Copyright 2024 The human_scene_transformer Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/human_scene_transformer/data/jrdb_train_detections_to_tracks.py b/human_scene_transformer/data/jrdb_train_detections_to_tracks.py
@@ -1,4 +1,4 @@
-# Copyright 2023 The human_scene_transformer Authors.
+# Copyright 2024 The human_scene_transformer Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/human_scene_transformer/data/math/data_types.py b/human_scene_transformer/data/math/data_types.py
@@ -1,4 +1,4 @@
-# Copyright 2023 The human_scene_transformer Authors.
+# Copyright 2024 The human_scene_transformer Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/human_scene_transformer/data/math/math_types.py b/human_scene_transformer/data/math/math_types.py
@@ -1,4 +1,4 @@
-# Copyright 2023 The human_scene_transformer Authors.
+# Copyright 2024 The human_scene_transformer Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/human_scene_transformer/data/math/pose3.py b/human_scene_transformer/data/math/pose3.py
@@ -1,4 +1,4 @@
-# Copyright 2023 The human_scene_transformer Authors.
+# Copyright 2024 The human_scene_transformer Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/human_scene_transformer/data/math/quaternion.py b/human_scene_transformer/data/math/quaternion.py
@@ -1,4 +1,4 @@
-# Copyright 2023 The human_scene_transformer Authors.
+# Copyright 2024 The human_scene_transformer Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -471,4 +471,9 @@ def __repr__(self) -> Text:
       Returns the string, 'Quaternion([x, y, z, w])', which can be used to
       regenerate the quaternion.
     """
-    return 'Quaternion([%r, %r, %r, %r])' % (self.x, self.y, self.z, self.w)
+    return 'Quaternion([%r, %r, %r, %r])' % (
+        float(self.x),
+        float(self.y),
+        float(self.z),
+        float(self.w),
+    )
diff --git a/human_scene_transformer/data/math/rotation3.py b/human_scene_transformer/data/math/rotation3.py
@@ -1,4 +1,4 @@
-# Copyright 2023 The human_scene_transformer Authors.
+# Copyright 2024 The human_scene_transformer Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/human_scene_transformer/data/math/vector_util.py b/human_scene_transformer/data/math/vector_util.py
@@ -1,4 +1,4 @@
-# Copyright 2023 The human_scene_transformer Authors.
+# Copyright 2024 The human_scene_transformer Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/human_scene_transformer/data/utils.py b/human_scene_transformer/data/utils.py
@@ -1,4 +1,4 @@
-# Copyright 2023 The human_scene_transformer Authors.
+# Copyright 2024 The human_scene_transformer Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/human_scene_transformer/is_hidden_generators.py b/human_scene_transformer/is_hidden_generators.py
@@ -1,4 +1,4 @@
-# Copyright 2023 The human_scene_transformer Authors.
+# Copyright 2024 The human_scene_transformer Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/human_scene_transformer/jrdb/dataset_params.py b/human_scene_transformer/jrdb/dataset_params.py
@@ -1,4 +1,4 @@
-# Copyright 2023 The human_scene_transformer Authors.
+# Copyright 2024 The human_scene_transformer Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/human_scene_transformer/jrdb/eval.py b/human_scene_transformer/jrdb/eval.py
@@ -1,4 +1,4 @@
-# Copyright 2023 The human_scene_transformer Authors.
+# Copyright 2024 The human_scene_transformer Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/human_scene_transformer/jrdb/eval_challenge.py b/human_scene_transformer/jrdb/eval_challenge.py
@@ -1,4 +1,4 @@
-# Copyright 2023 The human_scene_transformer Authors.
+# Copyright 2024 The human_scene_transformer Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/human_scene_transformer/jrdb/eval_keypoints.py b/human_scene_transformer/jrdb/eval_keypoints.py
@@ -1,4 +1,4 @@
-# Copyright 2023 The human_scene_transformer Authors.
+# Copyright 2024 The human_scene_transformer Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/human_scene_transformer/jrdb/input_fn.py b/human_scene_transformer/jrdb/input_fn.py
@@ -1,4 +1,4 @@
-# Copyright 2023 The human_scene_transformer Authors.
+# Copyright 2024 The human_scene_transformer Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/human_scene_transformer/jrdb/torch_dataset.py b/human_scene_transformer/jrdb/torch_dataset.py
@@ -1,4 +1,4 @@
-# Copyright 2023 The human_scene_transformer Authors.
+# Copyright 2024 The human_scene_transformer Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/human_scene_transformer/losses.py b/human_scene_transformer/losses.py
@@ -1,4 +1,4 @@
-# Copyright 2023 The human_scene_transformer Authors.
+# Copyright 2024 The human_scene_transformer Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/human_scene_transformer/metrics/ade.py b/human_scene_transformer/metrics/ade.py
@@ -1,4 +1,4 @@
-# Copyright 2023 The human_scene_transformer Authors.
+# Copyright 2024 The human_scene_transformer Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/human_scene_transformer/metrics/mean_angle_error.py b/human_scene_transformer/metrics/mean_angle_error.py
@@ -1,4 +1,4 @@
-# Copyright 2023 The human_scene_transformer Authors.
+# Copyright 2024 The human_scene_transformer Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/human_scene_transformer/metrics/metrics.py b/human_scene_transformer/metrics/metrics.py
@@ -1,4 +1,4 @@
-# Copyright 2023 The human_scene_transformer Authors.
+# Copyright 2024 The human_scene_transformer Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/human_scene_transformer/metrics/pos_nll.py b/human_scene_transformer/metrics/pos_nll.py
@@ -1,4 +1,4 @@
-# Copyright 2023 The human_scene_transformer Authors.
+# Copyright 2024 The human_scene_transformer Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/human_scene_transformer/model/agent_encoder.py b/human_scene_transformer/model/agent_encoder.py
@@ -1,4 +1,4 @@
-# Copyright 2023 The human_scene_transformer Authors.
+# Copyright 2024 The human_scene_transformer Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/human_scene_transformer/model/agent_feature_encoder.py b/human_scene_transformer/model/agent_feature_encoder.py
@@ -1,4 +1,4 @@
-# Copyright 2023 The human_scene_transformer Authors.
+# Copyright 2024 The human_scene_transformer Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/human_scene_transformer/model/agent_self_alignment.py b/human_scene_transformer/model/agent_self_alignment.py
@@ -1,4 +1,4 @@
-# Copyright 2023 The human_scene_transformer Authors.
+# Copyright 2024 The human_scene_transformer Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/human_scene_transformer/model/attention.py b/human_scene_transformer/model/attention.py
@@ -1,4 +1,4 @@
-# Copyright 2023 The human_scene_transformer Authors.
+# Copyright 2024 The human_scene_transformer Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/human_scene_transformer/model/embedding.py b/human_scene_transformer/model/embedding.py
@@ -1,4 +1,4 @@
-# Copyright 2023 The human_scene_transformer Authors.
+# Copyright 2024 The human_scene_transformer Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/human_scene_transformer/model/head.py b/human_scene_transformer/model/head.py
@@ -1,4 +1,4 @@
-# Copyright 2023 The human_scene_transformer Authors.
+# Copyright 2024 The human_scene_transformer Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/human_scene_transformer/model/model.py b/human_scene_transformer/model/model.py
@@ -1,4 +1,4 @@
-# Copyright 2023 The human_scene_transformer Authors.
+# Copyright 2024 The human_scene_transformer Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/human_scene_transformer/model/model_params.py b/human_scene_transformer/model/model_params.py
@@ -1,4 +1,4 @@
-# Copyright 2023 The human_scene_transformer Authors.
+# Copyright 2024 The human_scene_transformer Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/human_scene_transformer/model/multimodality_induction.py b/human_scene_transformer/model/multimodality_induction.py
@@ -1,4 +1,4 @@
-# Copyright 2023 The human_scene_transformer Authors.
+# Copyright 2024 The human_scene_transformer Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/human_scene_transformer/model/output_distributions.py b/human_scene_transformer/model/output_distributions.py
@@ -1,4 +1,4 @@
-# Copyright 2023 The human_scene_transformer Authors.
+# Copyright 2024 The human_scene_transformer Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/human_scene_transformer/model/preprocess.py b/human_scene_transformer/model/preprocess.py
@@ -1,4 +1,4 @@
-# Copyright 2023 The human_scene_transformer Authors.
+# Copyright 2024 The human_scene_transformer Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/human_scene_transformer/model/scene_encoder.py b/human_scene_transformer/model/scene_encoder.py
@@ -1,4 +1,4 @@
-# Copyright 2023 The human_scene_transformer Authors.
+# Copyright 2024 The human_scene_transformer Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/human_scene_transformer/pedestrians/dataset_params.py b/human_scene_transformer/pedestrians/dataset_params.py
@@ -1,4 +1,4 @@
-# Copyright 2023 The human_scene_transformer Authors.
+# Copyright 2024 The human_scene_transformer Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/human_scene_transformer/pedestrians/eval.py b/human_scene_transformer/pedestrians/eval.py
@@ -1,4 +1,4 @@
-# Copyright 2023 The human_scene_transformer Authors.
+# Copyright 2024 The human_scene_transformer Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/human_scene_transformer/pedestrians/input_fn.py b/human_scene_transformer/pedestrians/input_fn.py
@@ -1,4 +1,4 @@
-# Copyright 2023 The human_scene_transformer Authors.
+# Copyright 2024 The human_scene_transformer Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/human_scene_transformer/train.py b/human_scene_transformer/train.py
@@ -1,4 +1,4 @@
-# Copyright 2023 The human_scene_transformer Authors.
+# Copyright 2024 The human_scene_transformer Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/human_scene_transformer/train_model.py b/human_scene_transformer/train_model.py
@@ -1,4 +1,4 @@
-# Copyright 2023 The human_scene_transformer Authors.
+# Copyright 2024 The human_scene_transformer Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/human_scene_transformer/training_params.py b/human_scene_transformer/training_params.py
@@ -1,4 +1,4 @@
-# Copyright 2023 The human_scene_transformer Authors.
+# Copyright 2024 The human_scene_transformer Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.