[tapvid3d] Updates TAPVid-3D-ADT to RC3.

PiperOrigin-RevId: 657210406
Change-Id: If0d38d8bd5a570ffbea4d36d74629c05f5aa0b8d

tapnet/tapvid3d/annotation_generation/adt_utils.py

@@ -21,7 +21,6 @@
 tf.config.set_visible_devices([], "GPU")
 
 import os
-import hashlib
 
 import numpy as np
 import numpy.typing as npt
@@ -433,13 +432,11 @@ def process_vid(
         rgb_jpegs,
     )
 
-    # Verify trajectories to millimeter precision.
-    trajectories_hash = hashlib.md5(
-        np.round(trajectories * 1000).astype(np.int32).data.tobytes()
-    ).hexdigest()
-    assert trajectories_hash == in_npz["tracks_XYZ_hash"]
+    # Verify trajectories means.
+    trajectories_mean = trajectories.mean(axis=(0, 1))
+    assert np.allclose(trajectories_mean, in_npz["tracks_XYZ_mean"], atol=1e-5)
 
-    # Verify visibilities to 1e-5 precision.
+    # Verify visibilities means.
     assert np.abs(visibilities.mean() - in_npz["visibility_mean"]) < 1e-5
 
     example = {

tapnet/tapvid3d/annotation_generation/gcs_utils.py

@@ -27,7 +27,7 @@
 from tapnet.tapvid3d.splits import tapvid3d_splits  # pylint: disable=g-import-not-at-top, g-bad-import-order
 
 TAPVID3D_GCS_URL = (
-    "https://storage.googleapis.com/dm-tapnet/tapvid3d/release_files/rc2"
+    "https://storage.googleapis.com/dm-tapnet/tapvid3d/release_files/rc3"
 )