Test Dataset Multiplication by Timeshift

datasets/binar_masks.py

@@ -46,8 +46,10 @@ class BinaryMasksDataset(Dataset):
                     continue
                 filename, label = row.strip().split(',')
                 labeldict[filename] = self._to_label[label.lower()] if not self.setting == 'test' else filename
-        if self.stretch:
+        if self.stretch and self.setting == V.DATA_OPTIONS.train:
             labeldict.update({f'X_{key}': val for key, val in labeldict.items()})
+            labeldict.update({f'X_X_{key}': val for key, val in labeldict.items()})
+            labeldict.update({f'X_X_X_{key}': val for key, val in labeldict.items()})
         return labeldict
 
     def __len__(self):