| 12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061 |
- #!/usr/bin/env python
- import argparse
- import os
- import os.path as osp
- import shutil
- from glob import glob
- from tqdm import tqdm
- def get_subdir_name(src_path):
- basename = osp.basename(src_path)
- subdir_name, _ = osp.splitext(basename)
- return subdir_name
- def parse_args():
- parser = argparse.ArgumentParser()
- parser.add_argument(
- "--mode",
- default='copy',
- type=str,
- choices=['copy', 'link'],
- help="Copy or link images.")
- parser.add_argument(
- "--globs",
- nargs='+',
- type=str,
- help="Glob patterns used to find the images to be copied.")
- parser.add_argument(
- "--tags", nargs='+', type=str, help="Tags of each source directory.")
- parser.add_argument(
- "--save_dir",
- default='./',
- type=str,
- help="Path of directory to save collected results.")
- return parser.parse_args()
- if __name__ == '__main__':
- args = parse_args()
- if len(args.globs) != len(args.tags):
- raise ValueError(
- "The number of globs does not match the number of tags!")
- for pat, tag in zip(args.globs, args.tags):
- im_paths = glob(pat)
- print(f"Glob: {pat}\tTag: {tag}")
- for p in tqdm(im_paths):
- subdir_name = get_subdir_name(p)
- ext = osp.splitext(p)[1]
- subdir_path = osp.join(args.save_dir, subdir_name)
- subdir_path = osp.abspath(osp.normpath(subdir_path))
- if not osp.exists(subdir_path):
- os.makedirs(subdir_path)
- if args.mode == 'copy':
- shutil.copyfile(p, osp.join(subdir_path, tag + ext))
- elif args.mode == 'link':
- os.symlink(p, osp.join(subdir_path, tag + ext))
|
