From 08a6e072669d17cf54c6f59b3fcb47d1285c4e96 Mon Sep 17 00:00:00 2001 From: bro_SiN Date: Tue, 2 Apr 2024 19:51:39 +0800 Subject: [PATCH] Update sorting logic in scene/dataset_readers.py for readColmapSceneInfo --- scene/dataset_readers.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/scene/dataset_readers.py b/scene/dataset_readers.py index 2a6f904..b7a337c 100644 --- a/scene/dataset_readers.py +++ b/scene/dataset_readers.py @@ -143,7 +143,7 @@ def readColmapSceneInfo(path, images, eval, llffhold=8): reading_dir = "images" if images == None else images cam_infos_unsorted = readColmapCameras(cam_extrinsics=cam_extrinsics, cam_intrinsics=cam_intrinsics, images_folder=os.path.join(path, reading_dir)) - cam_infos = sorted(cam_infos_unsorted.copy(), key = lambda x : x.image_name) + cam_infos = sorted(cam_infos_unsorted.copy(), key = lambda x : x.uid) if eval: train_cam_infos = [c for idx, c in enumerate(cam_infos) if idx % llffhold != 0]