diff --git a/metadata/save_meta.py b/metadata/save_meta.py index 112e07c..8a83b9e 100644 --- a/metadata/save_meta.py +++ b/metadata/save_meta.py @@ -8,7 +8,7 @@ if __name__ == "__main__": parser.add_argument('--json_path', type=str) args = parser.parse_args() - fpath_list = sorted([os.path.join(args.data_path, fname) for fname in os.listdir(root) if fname.endswith('.pt')]) + fpath_list = sorted([os.path.join(args.data_path, fname) for fname in os.listdir(args.data_path) if fname.endswith('.pt')]) os.makedirs(args.json_path, exist_ok=True) json.dump(fpath_list, open(args.json_path, 'w'))