mirror
/
MeshDiffusion
kopia lustrzana https://github.com/lzzcd001/MeshDiffusion
1
0
Forkuj 0
Kod Zgłoszenia Packages Projekty Wydania Wiki Aktywność

Update save_meta.py

main
Zhen Liu 2023-08-02 02:14:39 +02:00 zatwierdzone przez GitHub
rodzic 45976e53db
commit 172af55945
Nie znaleziono w bazie danych klucza dla tego podpisu
ID klucza GPG: 4AEE18F83AFDEB23
1 zmienionych plików z 2 dodań i 1 usunięć
Pokaż statystyki Ściągnij plik aktualizacji Ściągnij plik porównania Expand all files Collapse all files

3
metadata/save_meta.py
Wyświetl plik

@ -6,9 +6,10 @@ if __name__ == "__main__":
parser = argparse.ArgumentParser()
parser.add_argument('--data_path', type=str)
parser.add_argument('--json_path', type=str)
parser.add_argument('--extension', type=str, default='pt')
args = parser.parse_args()
fpath_list = sorted([os.path.join(args.data_path, fname) for fname in os.listdir(args.data_path) if fname.endswith('.pt')])
fpath_list = sorted([os.path.join(args.data_path, fname[:-len(args.extension)-1]) for fname in os.listdir(args.data_path) if fname.endswith('.' + args.extension)])
os.makedirs(args.json_path, exist_ok=True)
json.dump(fpath_list, open(args.json_path, 'w'))