Lin Manhui 2 роки тому
батько
коміт
c5b8244a7b
1 змінених файлів з 2 додано та 2 видалено
  1. 2 2
      tools/prepare_dataset/common.py

+ 2 - 2
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