python path包的使用詳解

files函數的使用

import argparse
from path import Path

parser = argparse.ArgumentParser()

parser.add_argument("--dataset-dir", default='./infer_data/input', type=str, help="Dataset directory")
parser.add_argument("--img-exts", default=['png', 'jpg', 'bmp'], nargs='*', type=str, help="images extensions to glob")
args = parser.parse_args()
dataset_dir = Path(args.dataset_dir)
test_files = sum([dataset_dir.files('*.{}'.format(ext)) for ext in args.img_exts], [])
print(test_files)

# print_files: [Path('./infer_data/input/0000000023.png'), Path('./infer_data/input/0000000089.png'),...]
# [dataset_dir.files('*.{}'.format(ext)) for ext in args.img_exts]: [[Path('./infer_data/input/0000000023.png'),...], [], []]
# sum([[1, 2, 3],[]], []) :  [1, 2, 3]