Skip to content

Commit

Permalink
Merge remote-tracking branch 'Vkropt/patch-1'
Browse files Browse the repository at this point in the history
  • Loading branch information
FabianIsensee committed Feb 5, 2025
2 parents cd6b24f + 7359ea4 commit 55f6c50
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions nnunetv2/dataset_conversion/Dataset120_RoadSegmentation.py
Original file line number Diff line number Diff line change
Expand Up @@ -11,7 +11,7 @@
from scipy.ndimage import binary_fill_holes


def load_and_covnert_case(input_image: str, input_seg: str, output_image: str, output_seg: str,
def load_and_convert_case(input_image: str, input_seg: str, output_image: str, output_seg: str,
min_component_size: int = 50):
seg = io.imread(input_seg)
seg[seg == 255] = 1
Expand Down Expand Up @@ -55,7 +55,7 @@ def load_and_covnert_case(input_image: str, input_seg: str, output_image: str, o
for v in valid_ids:
r.append(
p.starmap_async(
load_and_covnert_case,
load_and_convert_case,
((
join(train_source, 'input', v),
join(train_source, 'output', v),
Expand All @@ -71,7 +71,7 @@ def load_and_covnert_case(input_image: str, input_seg: str, output_image: str, o
for v in valid_ids:
r.append(
p.starmap_async(
load_and_covnert_case,
load_and_convert_case,
((
join(test_source, 'input', v),
join(test_source, 'output', v),
Expand Down

0 comments on commit 55f6c50

Please sign in to comment.