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: when image name is 0 1 10 ...

Browse Source
This commit is contained in:
liuzhi 2024-05-13 21:42:34 +08:00
parent bfa79e356a
commit f9ef1ea404
2 changed files with 3 additions and 3 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_path.split('/')[-2], x.image_name))
cam_infos = sorted(cam_infos_unsorted.copy(), key = lambda x : (x.image_path.split('/')[-2], int(x.image_name)))
# 根据是否为评估模式（eval），将相机分为训练集和测试集
# 如果为评估模式，根据llffhold参数（通常用于LLFF数据集）间隔选择测试相机

4
train.py
View File

@ -230,10 +230,10 @@ if __name__ == "__main__":
parser.add_argument("--test_iterations", nargs="+", type=int, default=[7_000, 30_000])
parser.add_argument("--save_iterations", nargs="+", type=int, default=[7_000, 30_000])
parser.add_argument("--quiet", action="store_true")
parser.add_argument("--checkpoint_iterations", nargs="+", type=int, default=[])
parser.add_argument("--checkpoint_iterations", nargs="+", type=int, default=[30_000])
parser.add_argument("--start_checkpoint", type=str, default = None)
# 从命令行参数中解析出所有的参数值,并与上面的设置的参数一起存储到 Namespace 对象中，即args
# 从命令行参数中解析出所有的参数值,并与上面设置的参数一起存储到 Namespace 对象中，即args
args = parser.parse_args(sys.argv[1:])
args.save_iterations.append(args.iterations)