fix: update Label augmentation to include recompute targets

Author: torms3

deepem/data/augment/cortex/aug_16nm.py

@@ -90,6 +90,6 @@ def get_augmentation(
 
     # Recompute connected components
     if recompute:
-        augs.append(Label())
+        augs.append(Label(targets=recompute))
 
     return Compose(augs)

deepem/data/augment/cortex/aug_4nm.py

@@ -90,6 +90,6 @@ def get_augmentation(
 
     # Recompute connected components
     if recompute:
-        augs.append(Label())
+        augs.append(Label(targets=recompute))
 
     return Compose(augs)

deepem/data/augment/cortex/aug_8nm.py

@@ -90,6 +90,6 @@ def get_augmentation(
 
     # Recompute connected components
     if recompute:
-        augs.append(Label())
+        augs.append(Label(targets=recompute))
 
     return Compose(augs)

deepem/train/option.py

@@ -107,7 +107,7 @@ def initialize(self):
         self.parser.add_argument('--updown_scale_factor', type=vec3f, default=None)
 
         # Data augmentation
-        self.parser.add_argument('--recompute', action='store_true')
+        self.parser.add_argument('--recompute', type=str, default=[], nargs='+')
         self.parser.add_argument('--border', type=str, default=[], nargs='+')
         self.parser.add_argument('--flip', action='store_true')
         self.parser.add_argument('--grayscale', action='store_true')