From 7859f19f50736e477f0128a4c7b9942f41f597c0 Mon Sep 17 00:00:00 2001 From: Matthew Tancik Date: Fri, 16 Jun 2023 11:40:12 -0700 Subject: [PATCH] Fix data downscale paths (#2071) * Fix data downscale paths * ruff * format * Update process_data_utils.py --- nerfstudio/process_data/process_data_utils.py | 8 ++------ 1 file changed, 2 insertions(+), 6 deletions(-) diff --git a/nerfstudio/process_data/process_data_utils.py b/nerfstudio/process_data/process_data_utils.py index f8441b319..459b90e70 100644 --- a/nerfstudio/process_data/process_data_utils.py +++ b/nerfstudio/process_data/process_data_utils.py @@ -15,7 +15,6 @@ """Helper utils for processing data into the nerfstudio format.""" import math -import os import shutil import sys from enum import Enum @@ -348,11 +347,8 @@ def downscale_images( assert isinstance(downscale_factor, int) downscale_dir = image_dir.parent / f"{folder_name}_{downscale_factor}" downscale_dir.mkdir(parents=True, exist_ok=True) - # Using %05d ffmpeg commands appears to be unreliable (skips images), so use scandir. - files = os.scandir(image_dir) - for f in files: - if f.is_dir(): - continue + # Using %05d ffmpeg commands appears to be unreliable (skips images). + for f in list_images(image_dir): filename = f.name nn_flag = "" if not nearest_neighbor else ":flags=neighbor" ffmpeg_cmd = [