Pytorch3d: Change Color of Mesh Vertices? with torch.ones_like(verts)[None]?

[karim-eladl]

Hi! I have a super quick question but somehow I can't figure it out based on the tutorials and forums.

I'm rendering a set of objects that are saved as one set of vertices and faces but are not connected. I'd like to set one of these objects to be not white.

I'd use verts_rgb = torch.ones_like(verts[0:100])[None] to set vertices 0 through 100 to white. How do I set say verts[101:200] to another color?

Thank you so much!

[bottler]

Do you want something like

verts_rgb = torch.ones_like(verts)[None]
verts_rgb[0,101:200,:]=torch.tensor([red_level, green_level, blue_level])

[karim-eladl]

yes thank you!