Depth render output shape

[2454511550Lin]

Thank you for the amazing work. I encounter a problem regarding the shape of the rendered depth map. I tried to retrieve the rendered_image and rendered_depth by running:

rendered_image, radii, rendered_depth, rendered_alpha = rasterizer(...)

I expect rendered_image has the shape of (3, Width, Hight), and rendered_depth has the shape of (1, Width, Hight).

However, what I got is (3, 719, 1279) for rendered_image (which is correct regarding my image resolution), but (1536351,) for the rendered_depth. I am wondering why rendered_depth is a flat vector? Also, I cannot reshape it to any H x W that makes sense to my scene. If other people does not got a flat vector but (1, Width, Hight), please let me know.

Any help or suggestions would be greatly appreciated!!

[2454511550Lin]

I realize that my rendered_depth is of the same size as the number of Gaussian spheres in the scene. Can someone help me understand what it actually stands for? It is the depth of all the Gaussian spheres under the current camera view? Thanks a lot!