Skip to content

Commit b9c4488

Browse files
TensorFlow Datasets Teamcopybara-github
authored andcommitted
Fix nondeterminism in splitting for Kitti.
PiperOrigin-RevId: 258300182
1 parent 50a6921 commit b9c4488

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

tensorflow_datasets/image/kitti.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -72,7 +72,7 @@
7272
class Kitti(tfds.core.GeneratorBasedBuilder):
7373
"""Kitti dataset."""
7474

75-
VERSION = tfds.core.Version("3.0.0")
75+
VERSION = tfds.core.Version("3.1.0")
7676
SUPPORTED_VERSIONS = [
7777
tfds.core.Version("2.0.0"),
7878
]
@@ -289,10 +289,10 @@ def _build_splits(devkit):
289289
_VALIDATION_SPLIT_PERCENT_VIDEOS * len(video_to_image) // 100)
290290
test_videos = set(
291291
np.random.choice(
292-
list(video_to_image.keys()), num_test_videos, replace=False))
292+
sorted(list(video_to_image.keys())), num_test_videos, replace=False))
293293
validation_videos = set(
294294
np.random.choice(
295-
list(set(video_to_image.keys()) - set(test_videos)),
295+
sorted(list(set(video_to_image.keys()) - set(test_videos))),
296296
num_validation_videos,
297297
replace=False))
298298
test_images = []

0 commit comments

Comments
 (0)