tink/nerfstudio
4
0 Fork 0
mirror of https://github.com/nerfstudio-project/nerfstudio.git synced 2026-11-15 04:13:57 +08:00
Code Activity

Fix progress bar for full images datamanager

Browse Source
This commit is contained in:
Brent Yi committed 2024-02-28 05:05:46 -08:00
1 parent c491e3e101
commit f9ae770f34
1 file changed
+15 -11
Download Patch File Download Diff File Expand all files Collapse all files
nerfstudio/data/datamanagers/full_images_datamanager.py
+15 -11
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),
)
)