From 082a0206b4cdc2e7a295612cb044f2f1c4fcc75b Mon Sep 17 00:00:00 2001 From: Yi Yang Date: Thu, 6 Jun 2024 14:38:52 +0100 Subject: [PATCH] Fix kubric data loading of video shape (None, 256, 256, None) (#327) --- challenges/point_tracking/dataset.py | 1 + 1 file changed, 1 insertion(+) diff --git a/challenges/point_tracking/dataset.py b/challenges/point_tracking/dataset.py index c08c66a5..0f4aca3d 100644 --- a/challenges/point_tracking/dataset.py +++ b/challenges/point_tracking/dataset.py @@ -929,6 +929,7 @@ def add_tracks(data, ) video = tf.slice(video, start, size) video = tf.image.resize(tf.cast(video, tf.float32), train_size) + video.set_shape([num_frames, train_size[0], train_size[1], 3]) if vflip: video = video[:, ::-1, :, :] target_points = target_points * np.array([1, -1])