UniqAI/gaussian-splatting
1
0
Fork 0
mirror of https://github.com/graphdeco-inria/gaussian-splatting synced 2025-03-29 03:43:27 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix: multi subfolder

Browse Source
This commit is contained in:
liuzhi 2024-05-11 15:55:59 +08:00
parent 1dd0a100b5
commit a0534cfb64
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
scene/dataset_readers.py
View File

@ -173,7 +173,7 @@ def readColmapSceneInfo(path, images, eval, llffhold=8):
# 读取并处理相机参数，转换为内部使用的格式
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.image_path.split('/')[-2], x.image_name))
# 根据是否为评估模式（eval），将相机分为训练集和测试集
# 如果为评估模式，根据llffhold参数（通常用于LLFF数据集）间隔选择测试相机