You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
42 lines
1.3 KiB
42 lines
1.3 KiB
#!/usr/bin/env python |
|
|
|
import os.path as osp |
|
|
|
from common import (get_default_parser, add_crop_options, crop_patches, |
|
get_path_tuples, create_file_list, link_dataset) |
|
|
|
SUBSETS = ('train', 'val', 'test') |
|
SUBDIRS = ('A', 'B', 'label') |
|
FILE_LIST_PATTERN = "{subset}.txt" |
|
URL = "" |
|
|
|
if __name__ == '__main__': |
|
parser = get_default_parser() |
|
parser = add_crop_options(parser) |
|
args = parser.parse_args() |
|
|
|
out_dir = osp.join(args.out_dataset_dir, |
|
osp.basename(osp.normpath(args.in_dataset_dir))) |
|
|
|
if args.crop_size is not None: |
|
crop_patches( |
|
args.crop_size, |
|
args.crop_stride, |
|
data_dir=args.in_dataset_dir, |
|
out_dir=out_dir, |
|
subsets=SUBSETS, |
|
subdirs=SUBDIRS, |
|
glob_pattern='*.png', |
|
max_workers=0) |
|
else: |
|
link_dataset(args.in_dataset_dir, args.out_dataset_dir) |
|
|
|
for subset in SUBSETS: |
|
path_tuples = get_path_tuples( |
|
*(osp.join(out_dir, subset, subdir) for subdir in SUBDIRS), |
|
glob_pattern='**/*.png', |
|
data_dir=args.out_dataset_dir) |
|
file_list = osp.join( |
|
args.out_dataset_dir, FILE_LIST_PATTERN.format(subset=subset)) |
|
create_file_list(file_list, path_tuples) |
|
print(f"Write file list to {file_list}.")
|
|
|