fix erosion option bug

Author: donglaiw

connectomics/data/utils/data_transform.py

@@ -61,7 +61,7 @@ def edt_instance(label: np.ndarray,
                  mode: str = '2d',
                  quantize: bool = True,
                  resolution: Tuple[float] = (1.0, 1.0, 1.0),
-                 padding: bool = False):
+                 padding: bool = False),
                  erosion: int = 0):
     assert mode in ['2d', '3d']
     if mode == '3d':

connectomics/model/utils/misc.py

@@ -125,7 +125,7 @@ def __init__(self,
             if topt[0] == '5': # instance_edt
                 if len(topt) == 1:
                     topt = topt + '-2d-0-0-5.0' # 2d w/o padding or quantize
-                _, mode, padding, quant, z_res = topt.split('-')
+                _, mode, padding, quant, z_res, erosion = topt.split('-')
                 if bool(int(quant)): # quantized by 0.1 bin (additional one for bg)
                     self.split_channels.append(11)
                     continue