From da57d3f5ba5362391b961cb4ce8b5eda4e97268f Mon Sep 17 00:00:00 2001 From: "J.Y" <132313008+jb-ye@users.noreply.github.com> Date: Thu, 18 Jan 2024 17:15:05 -0500 Subject: [PATCH] MAX_AUTO_RESOLUTION is ensured in auto downscale (#2788) Co-authored-by: Jianbo Ye --- nerfstudio/data/dataparsers/colmap_dataparser.py | 2 +- nerfstudio/data/dataparsers/nerfstudio_dataparser.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/nerfstudio/data/dataparsers/colmap_dataparser.py b/nerfstudio/data/dataparsers/colmap_dataparser.py index a0ee580e35..ce3cf8919f 100644 --- a/nerfstudio/data/dataparsers/colmap_dataparser.py +++ b/nerfstudio/data/dataparsers/colmap_dataparser.py @@ -486,7 +486,7 @@ def get_fname(parent: Path, filepath: Path) -> Path: max_res = max(h, w) df = 0 while True: - if (max_res / 2 ** (df)) < MAX_AUTO_RESOLUTION: + if (max_res / 2 ** (df)) <= MAX_AUTO_RESOLUTION: break df += 1 diff --git a/nerfstudio/data/dataparsers/nerfstudio_dataparser.py b/nerfstudio/data/dataparsers/nerfstudio_dataparser.py index 554e88dac0..2d007d8e9a 100644 --- a/nerfstudio/data/dataparsers/nerfstudio_dataparser.py +++ b/nerfstudio/data/dataparsers/nerfstudio_dataparser.py @@ -376,7 +376,7 @@ def _get_fname(self, filepath: Path, data_dir: Path, downsample_folder_prefix="i max_res = max(h, w) df = 0 while True: - if (max_res / 2 ** (df)) < MAX_AUTO_RESOLUTION: + if (max_res / 2 ** (df)) <= MAX_AUTO_RESOLUTION: break if not (data_dir / f"{downsample_folder_prefix}{2**(df+1)}" / filepath.name).exists(): break