Skip to content

Commit 4cac781

Browse files
committed
feat(option): binarize blood vessel label
1 parent 1a60c74 commit 4cac781

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

deepem/train/option.py

+3
Original file line numberDiff line numberDiff line change
@@ -289,6 +289,9 @@ def parse(self):
289289
"soma",
290290
]
291291

292+
if opt.blv_num_channels == 1:
293+
requires_binarize.append("blood_vessel")
294+
292295
# Test training
293296
if opt.test:
294297
opt.eval_intv = 100

0 commit comments

Comments
 (0)