UniqAI/gaussian-splatting
1
0
Fork 0
mirror of https://github.com/graphdeco-inria/gaussian-splatting synced 2025-05-07 13:34:26 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Add assert check for valid point clouds for COLMAP dataset

Browse Source 
This addresses error `RuntimeError: CUDA error: invalid configuration argument` due to zero points
This commit is contained in:
Nandan Manjunatha 2025-04-27 16:36:26 +05:30 committed by GitHub
parent 54c035f783
commit ef57af14df
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
scene/dataset_readers.py
View File

@ -211,6 +211,7 @@ def readColmapSceneInfo(path, images, depths, eval, train_test_exp, llffhold=8):
xyz, rgb, _ = read_points3D_binary(bin_path) xyz, rgb, _ = read_points3D_binary(bin_path)
except: except:
xyz, rgb, _ = read_points3D_text(txt_path) xyz, rgb, _ = read_points3D_text(txt_path)
assert xyz and rgb, f"Error loading {path} data. Point data must exist in either points3D.bin or points3D.txt"
storePly(ply_path, xyz, rgb) storePly(ply_path, xyz, rgb)
try: try:
pcd = fetchPly(ply_path) pcd = fetchPly(ply_path)
@ -312,4 +313,4 @@ def readNerfSyntheticInfo(path, white_background, depths, eval, extension=".png"
sceneLoadTypeCallbacks = { sceneLoadTypeCallbacks = {
"Colmap": readColmapSceneInfo, "Colmap": readColmapSceneInfo,
"Blender" : readNerfSyntheticInfo "Blender" : readNerfSyntheticInfo
} }