From dcbd26dcea42baf819c69e3809807f1e53636f0e Mon Sep 17 00:00:00 2001 From: Jachin Date: Mon, 2 Oct 2023 15:57:32 -0400 Subject: [PATCH] allow for subfolders in images dir --- scene/dataset_readers.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/scene/dataset_readers.py b/scene/dataset_readers.py index 2a6f904..080f4a5 100644 --- a/scene/dataset_readers.py +++ b/scene/dataset_readers.py @@ -94,7 +94,7 @@ def readColmapCameras(cam_extrinsics, cam_intrinsics, images_folder): else: assert False, "Colmap camera model not handled: only undistorted datasets (PINHOLE or SIMPLE_PINHOLE cameras) supported!" - image_path = os.path.join(images_folder, os.path.basename(extr.name)) + image_path = os.path.join(images_folder, extr.name) image_name = os.path.basename(image_path).split(".")[0] image = Image.open(image_path) @@ -257,4 +257,4 @@ def readNerfSyntheticInfo(path, white_background, eval, extension=".png"): sceneLoadTypeCallbacks = { "Colmap": readColmapSceneInfo, "Blender" : readNerfSyntheticInfo -} \ No newline at end of file +}