tink/nerfstudio
4
0
Fork 0
mirror of https://github.com/nerfstudio-project/nerfstudio.git synced 2026-11-11 02:24:38 +08:00
Code

Missing images error warning (#1122)

Browse Source
This commit is contained in:
Matthew Tancik 2022-12-13 13:24:22 -08:00 committed by GitHub
parent 96f269cc1a
commit c8daa1a1d7
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
nerfstudio/process_data/process_data_utils.py
View File

@ -167,6 +167,10 @@ def copy_images(data: Path, image_dir: Path, verbose) -> int:
allowed_exts = [".jpg", ".jpeg", ".png", ".tif", ".tiff"]
image_paths = sorted([p for p in data.glob("[!.]*") if p.suffix.lower() in allowed_exts])
if len(image_paths) == 0:
CONSOLE.log("[bold red]:skull: No usable images in the data folder.")
sys.exit(1)
num_frames = len(copy_images_list(image_paths, image_dir, verbose))
return num_frames