Update common.py

own
Lin Manhui 3 years ago committed by GitHub
parent fc1613b4a4
commit c5b8244a7b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
  1. 4
      tools/prepare_dataset/common.py

@ -107,7 +107,7 @@ def crop_patches(crop_size,
if max_workers < 0: if max_workers < 0:
raise ValueError("`max_workers` must be a non-negative integer!") raise ValueError("`max_workers` must be a non-negative integer!")
if subset is None: if subsets is None:
subsets = ('', ) subsets = ('', )
if max_workers == 0: if max_workers == 0:
@ -280,4 +280,4 @@ def random_split(samples,
# Append remainder to the last split # Append remainder to the last split
splits[-1].append(splits[ed_idx:]) splits[-1].append(splits[ed_idx:])
return splits return splits

Loading…
Cancel
Save