adjust first test image start idx

ziyc · Oct 24, 2024 · 641aa88 · 641aa88
1 parent 388642a
commit 641aa88
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/datasets/driving_dataset.py b/datasets/driving_dataset.py
@@ -585,7 +585,7 @@ def split_train_test(self):
         if self.data_cfg.pixel_source.test_image_stride != 0:
             test_timesteps = np.arange(
                 # it makes no sense to have test timesteps before the start timestep
-                self.data_cfg.pixel_source.test_image_stride,
+                max(0,self.data_cfg.pixel_source.test_image_stride-1),
                 self.num_img_timesteps,
                 self.data_cfg.pixel_source.test_image_stride,
             )