|
|
@@ -20,17 +20,18 @@ from .dataset_split.imagenet_split import split_imagenet_dataset
|
|
|
from .dataset_split.seg_split import split_seg_dataset
|
|
|
|
|
|
|
|
|
-def dataset_split(dataset_dir, dataset_form, val_value, test_value, save_dir):
|
|
|
- if dataset_form == "coco":
|
|
|
+def dataset_split(dataset_dir, dataset_format, val_value, test_value,
|
|
|
+ save_dir):
|
|
|
+ if dataset_format == "coco":
|
|
|
train_num, val_num, test_num = split_coco_dataset(
|
|
|
dataset_dir, val_value, test_value, save_dir)
|
|
|
- elif dataset_form == "voc":
|
|
|
+ elif dataset_format == "voc":
|
|
|
train_num, val_num, test_num = split_voc_dataset(
|
|
|
dataset_dir, val_value, test_value, save_dir)
|
|
|
- elif dataset_form == "seg":
|
|
|
+ elif dataset_format == "seg":
|
|
|
train_num, val_num, test_num = split_seg_dataset(
|
|
|
dataset_dir, val_value, test_value, save_dir)
|
|
|
- elif dataset_form == "imagenet":
|
|
|
+ elif dataset_format == "imagenet":
|
|
|
train_num, val_num, test_num = split_imagenet_dataset(
|
|
|
dataset_dir, val_value, test_value, save_dir)
|
|
|
print("Dataset Split Done.")
|