From c8daa1a1d756128a2bbdba4ee39d49ffbf4a2325 Mon Sep 17 00:00:00 2001 From: Matthew Tancik Date: Tue, 13 Dec 2022 13:24:22 -0800 Subject: [PATCH] Missing images error warning (#1122) --- nerfstudio/process_data/process_data_utils.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/nerfstudio/process_data/process_data_utils.py b/nerfstudio/process_data/process_data_utils.py index e291d898a..e028bf1c0 100644 --- a/nerfstudio/process_data/process_data_utils.py +++ b/nerfstudio/process_data/process_data_utils.py @@ -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