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.
|
|
|
|
#!/usr/bin/env python
|
|
|
|
|
|
|
|
|
|
import os.path as osp
|
|
|
|
|
|
|
|
|
|
from common import (get_default_parser, get_path_tuples, create_file_list,
|
|
|
|
|
link_dataset)
|
|
|
|
|
|
|
|
|
|
SUBSETS = ('train', 'val', 'test')
|
|
|
|
|
SUBDIRS = ('A', 'B', 'OUT')
|
|
|
|
|
FILE_LIST_PATTERN = "{subset}.txt"
|
|
|
|
|
URL = ""
|
|
|
|
|
|
|
|
|
|
if __name__ == '__main__':
|
|
|
|
|
parser = get_default_parser()
|
|
|
|
|
args = parser.parse_args()
|
|
|
|
|
|
|
|
|
|
out_dir = osp.join(args.out_dataset_dir,
|
|
|
|
|
osp.basename(osp.normpath(args.in_dataset_dir)))
|
|
|
|
|
|
|
|
|
|
link_dataset(args.in_dataset_dir, args.out_dataset_dir)
|
|
|
|
|
|
|
|
|
|
for subset in SUBSETS:
|
|
|
|
|
# NOTE: Only use cropped real samples.
|
|
|
|
|
path_tuples = get_path_tuples(
|
|
|
|
|
*(osp.join(out_dir, 'Real', 'subset', subset, subdir)
|
|
|
|
|
for subdir in SUBDIRS),
|
|
|
|
|
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}.")
|
