模仿COLMAP的数据跑3DGS（自用）

一条葱花雩

已于 2024-07-16 16:19:41 修改

阅读量686

点赞数 6

文章标签：计算机视觉

于 2024-07-16 16:13:57 首次发布

本文链接：https://blog.csdn.net/m0_53406928/article/details/140334823

版权

首先需要搞清楚3DGS的数据类型要求是什么

3DGS/gaussian-splatting/scene/dataset_readers.py里面

sceneLoadTypeCallbacks = {
    "Colmap": readColmapSceneInfo,
    "Blender" : readNerfSyntheticInfo
}

一个是colmap的场景读取一个是合成场景的数据读取。

先看colmap的场景：

def readColmapSceneInfo(path, images, eval, llffhold=8):
    try:
        cameras_extrinsic_file = os.path.join(path, "sparse/0", "images.bin")
        cameras_intrinsic_file = os.path.join(path, "sparse/0", "cameras.bin")
        cam_extrinsics = read_extrinsics_binary(cameras_extrinsic_file)
        cam_intrinsics = read_intrinsics_binary(cameras_intrinsic_file)
    except:
        cameras_extrinsic_file = os.path.join(path, "sparse/0", "images.txt")
        cameras_intrinsic_file = os.path.join(path, "sparse/0", "cameras.txt")
        cam_extrinsics = read_extrinsics_text(cameras_extrinsic_file)
        cam_intrinsics = read_intrinsics_text(cameras_intrinsic_file)

    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, r