From 2b7a908cf8b13bc7b236f0afa2a1ba5928e64f3d Mon Sep 17 00:00:00 2001 From: Zhen Liu Date: Thu, 20 Apr 2023 11:07:22 +0200 Subject: [PATCH] Update save_meta.py --- metadata/save_meta.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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'))