tink/nerfstudio
4
0
Fork 0
mirror of https://github.com/nerfstudio-project/nerfstudio.git synced 2026-11-11 10:34:29 +08:00
Code

Fix progress bar for full images datamanager (#2969)

Browse Source
This commit is contained in:
Brent Yi 2024-02-28 11:41:03 -08:00 committed by GitHub
parent c491e3e101
commit 8b728d4ec8
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 15 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
nerfstudio/data/datamanagers/full_images_datamanager.py
View File

@ -180,26 +180,30 @@ class FullImageDatamanager(DataManager, Generic[TDataset]):
return data
CONSOLE.log("Caching / undistorting train images")
with ThreadPoolExecutor() as executor:
with ThreadPoolExecutor(max_workers=2) as executor:
cached_train = list(
executor.map(
process_train_data,
track(
track(
executor.map(
process_train_data,
range(len(self.train_dataset)),
description="Caching / undistorting train images",
transient=True,
),
description="Caching / undistorting train images",
transient=True,
total=len(self.train_dataset),
)
)
CONSOLE.log("Caching / undistorting eval images")
with ThreadPoolExecutor() as executor:
with ThreadPoolExecutor(max_workers=2) as executor:
cached_eval = list(
executor.map(
process_eval_data,
track(
range(len(self.eval_dataset)), description="Caching / undistorting eval images", transient=True
track(
executor.map(
process_eval_data,
range(len(self.eval_dataset)),
),
description="Caching / undistorting eval images",
transient=True,
total=len(self.eval_dataset),
)
)