Update save_meta.py

main
Zhen Liu 2023-04-20 11:07:22 +02:00 zatwierdzone przez GitHub
rodzic e3ace6b510
commit 2b7a908cf8
Nie znaleziono w bazie danych klucza dla tego podpisu
ID klucza GPG: 4AEE18F83AFDEB23
1 zmienionych plików z 1 dodań i 1 usunięć

Wyświetl plik

@ -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'))