steffen/masks_augments_compare-21
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Test Dataset Multiplication by Timeshift

Browse Source
This commit is contained in:
Si11ium
2020-05-12 16:18:02 +02:00
parent 28bfcfdce3
commit dce799a52b
4 changed files with 9 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
datasets/binar_masks.py
View File

@ -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):