backup/shap-e
1
0
Fork 0
mirror of https://github.com/openai/shap-e.git synced 2026-02-03 02:04:33 +08:00
Code Issues Packages Projects Releases Wiki Activity

format with black

Browse Source
This commit is contained in:
Alex Nichol
2023-05-08 17:21:21 -07:00
parent 64304879ff
commit 5fad78ad78
1 changed files with 2 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
shap_e/rendering/mesh.py
View File

@@ -89,17 +89,13 @@ class TriMesh:
def write_obj(self, raw_f: BinaryIO):
if self.has_vertex_colors():
vertex_colors = np.stack([self.vertex_channels[x]
for x in "RGB"], axis=1)
vertex_colors = np.stack([self.vertex_channels[x] for x in "RGB"], axis=1)
vertices = [
"{} {} {} {} {} {}".format(*coord, *color)
for coord, color in zip(self.verts.tolist(), vertex_colors.tolist())
]
else:
vertices = [
"{} {} {}".format(*coord)
for coord in self.verts.tolist()
]
vertices = ["{} {} {}".format(*coord) for coord in self.verts.tolist()]
faces = [
"f {} {} {}".format(str(tri[0] + 1), str(tri[1] + 1), str(tri[2] + 1))