From 641aa8899796afe80ed835ef1ccbce13b75435b5 Mon Sep 17 00:00:00 2001 From: Georg Hess Date: Thu, 24 Oct 2024 08:22:35 +0200 Subject: [PATCH] adjust first test image start idx --- datasets/driving_dataset.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/datasets/driving_dataset.py b/datasets/driving_dataset.py index 38b3e07..15e8da5 100644 --- 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, )