speed aug fixed

2020-05-21 14:42:35 +02:00
parent d58bcbf14b
commit e4abbc045c
3 changed files with 15 additions and 17 deletions
--- a/datasets/binar_masks.py
+++ b/datasets/binar_masks.py
@ -51,7 +51,6 @@ class BinaryMasksDataset(Dataset):
            additional_dict = ({f'X{key}': val for key, val in labeldict.items()})
            additional_dict.update({f'XX{key}': val for key, val in labeldict.items()})
            additional_dict.update({f'XXX{key}': val for key, val in labeldict.items()})
-            additional_dict.update({f'XXXX{key}': val for key, val in labeldict.items()})
            labeldict.update(additional_dict)

        # Delete File if one exists.