Merge branch 'main' of github.com:xing-lab-pitt/livecellx

xing-lab-pitt · Nov 1, 2024 · 71241ed · 71241ed
2 parents 752b28e + 34b71c2
commit 71241ed
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/livecellx/model_zoo/segmentation/train_csn.py b/livecellx/model_zoo/segmentation/train_csn.py
@@ -46,7 +46,7 @@ def parse_args():
         dest="input_type",
         type=str,
         default="raw_aug_seg",
-        choices=["raw_aug_seg", "raw_aug_duplicate", "raw_duplicate", "edt_v0"],
+        choices=["raw_aug_seg", "raw_aug_duplicate", "raw_duplicate", "edt_v0", "edt_v1"],
     )
     parser.add_argument("--apply_gt_seg_edt", dest="apply_gt_seg_edt", default=False, action="store_true")
     parser.add_argument("--class-weights", dest="class_weights", type=str, default="1,1,1")