From 34b71c2b4c2c3e681d38ef7b42f8bedaeafdfc9d Mon Sep 17 00:00:00 2001 From: Ke Date: Tue, 15 Oct 2024 14:48:32 -0400 Subject: [PATCH] feat: add 'edt_v1' as a valid input type in train_csn.py --- livecellx/model_zoo/segmentation/train_csn.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/livecellx/model_zoo/segmentation/train_csn.py b/livecellx/model_zoo/segmentation/train_csn.py index 5ee9603..eec8310 100644 --- 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")