fix data path

transfuser/config.py

@@ -11,10 +11,10 @@ class GlobalConfig:
     val_towns = ['Town05']
     train_data, val_data = [], []
     for town in train_towns:
-        train_data.append(os.path.join(root_dir, town+''))
-        train_data.append(os.path.join(root_dir, town+'_small'))
+        train_data.append(os.path.join(root_dir, town+'_tiny'))
+        train_data.append(os.path.join(root_dir, town+'_short'))
     for town in val_towns:
-        val_data.append(os.path.join(root_dir, town+''))
+        val_data.append(os.path.join(root_dir, town+'_short'))
 
     ignore_sides = True # don't consider side cameras
     ignore_rear = True # don't consider rear cameras

transfuser/data.py

@@ -37,12 +37,7 @@ class CARLA_Data(Dataset):
         self.velocity = []
         
         for sub_root in tqdm(root, file=sys.stdout):
-            print("Preload subroot", flush=True)
-            town = os.path.basename(os.path.normpath(sub_root))
-            print("Town: ", town, flush=True)
-            writefolder = '/mnt/qb/geiger/bjaeger25' #Workaround since I don't have write permissions in the dataset folder
-            preload_file = os.path.join(writefolder, str(town) + '_rg_lidar_diag_pl_'+str(self.seq_len)+'_'+str(self.pred_len)+'.npy')
-            print("Preload_file: ", preload_file, flush=True)
+            preload_file = os.path.join(sub_root, 'rg_lidar_diag_pl_'+str(self.seq_len)+'_'+str(self.pred_len)+'.npy')
 
             # dump to npy if no preload
             if not os.path.exists(preload_file):