UniqAI/gaussian-splatting
1
0
Fork 0
mirror of https://github.com/graphdeco-inria/gaussian-splatting synced 2024-11-22 00:08:02 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Bump rasterizer, faster text loading

Browse Source
This commit is contained in:
bkerbl 2023-09-05 19:30:53 +02:00
parent b2ada78a77
commit ea68bdf29c
2 changed files with 21 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
scene/colmap_loader.py
View File

@ -89,6 +89,21 @@ def read_points3D_text(path):
xyzs = None xyzs = None
rgbs = None rgbs = None
errors = None errors = None
num_points = 0
with open(path, "r") as fid:
while True:
line = fid.readline()
if not line:
break
line = line.strip()
if len(line) > 0 and line[0] != "#":
num_points += 1
xyzs = np.empty((num_points, 3))
rgbs = np.empty((num_points, 3))
errors = np.empty((num_points, 1))
count = 0
with open(path, "r") as fid: with open(path, "r") as fid:
while True: while True:
line = fid.readline() line = fid.readline()
@ -100,14 +115,11 @@ def read_points3D_text(path):
xyz = np.array(tuple(map(float, elems[1:4]))) xyz = np.array(tuple(map(float, elems[1:4])))
rgb = np.array(tuple(map(int, elems[4:7]))) rgb = np.array(tuple(map(int, elems[4:7])))
error = np.array(float(elems[7])) error = np.array(float(elems[7]))
if xyzs is None: xyzs[count] = xyz
xyzs = xyz[None, ...] rgbs[count] = rgb
rgbs = rgb[None, ...] errors[count] = error
errors = error[None, ...] count += 1
else:
xyzs = np.append(xyzs, xyz[None, ...], axis=0)
rgbs = np.append(rgbs, rgb[None, ...], axis=0)
errors = np.append(errors, error[None, ...], axis=0)
return xyzs, rgbs, errors return xyzs, rgbs, errors
def read_points3D_binary(path_to_model_file): def read_points3D_binary(path_to_model_file):

2
submodules/diff-gaussian-rasterization

@ -1 +1 @@
Subproject commit 8064f52ca233942bdec2d1a1451c026deedd320b Subproject commit 59f5f77e3ddbac3ed9db93ec2cfe99ed6c5d121d