File tree Expand file tree Collapse file tree 1 file changed +3
-3
lines changed
tensorflow_datasets/image Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change 72
72
class Kitti (tfds .core .GeneratorBasedBuilder ):
73
73
"""Kitti dataset."""
74
74
75
- VERSION = tfds .core .Version ("3.0 .0" )
75
+ VERSION = tfds .core .Version ("3.1 .0" )
76
76
SUPPORTED_VERSIONS = [
77
77
tfds .core .Version ("2.0.0" ),
78
78
]
@@ -289,10 +289,10 @@ def _build_splits(devkit):
289
289
_VALIDATION_SPLIT_PERCENT_VIDEOS * len (video_to_image ) // 100 )
290
290
test_videos = set (
291
291
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 ))
293
293
validation_videos = set (
294
294
np .random .choice (
295
- list (set (video_to_image .keys ()) - set (test_videos )),
295
+ sorted ( list (set (video_to_image .keys ()) - set (test_videos ) )),
296
296
num_validation_videos ,
297
297
replace = False ))
298
298
test_images = []
You can’t perform that action at this time.
0 commit comments