2023-03-11 10:17:59 +00:00
|
|
|
import os
|
|
|
|
import json
|
|
|
|
import argparse
|
|
|
|
|
|
|
|
if __name__ == "__main__":
|
|
|
|
parser = argparse.ArgumentParser()
|
|
|
|
parser.add_argument('--data_path', type=str)
|
|
|
|
parser.add_argument('--json_path', type=str)
|
2023-08-02 00:14:39 +00:00
|
|
|
parser.add_argument('--extension', type=str, default='pt')
|
2023-03-11 10:17:59 +00:00
|
|
|
args = parser.parse_args()
|
|
|
|
|
2023-08-02 00:17:28 +00:00
|
|
|
fpath_list = sorted([os.path.join(args.data_path, fname) for fname in os.listdir(args.data_path) if fname.endswith('.' + args.extension)])
|
2023-03-11 10:17:59 +00:00
|
|
|
json.dump(fpath_list, open(args.json_path, 'w'))
|
|
|
|
|