small change when FILTER_VALUE >1

Doodleverse · Oct 20, 2022 · 1ccde41 · 1ccde41
1 parent c9b3f9d
commit 1ccde41
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/make_nd_dataset.py b/make_nd_dataset.py
@@ -648,7 +648,7 @@ def read_seg_dataset_multiclass(example):
                     l = remove_small_objects(lstack[:,:,kk].astype('uint8')>0, np.pi*(FILTER_VALUE**2))
                     l = remove_small_holes(lstack[:,:,kk].astype('uint8')>0, np.pi*(FILTER_VALUE**2))
                     lstack[:,:,kk] = np.round(l).astype(np.uint8)
-                    del l
+                    #del l
 
             datadict={}
             datadict['arr_0'] = im.astype(np.uint8)