Add full app

app.py

@@ -24,8 +24,8 @@ def generate(*args):
             im.save(p.stdin, "PNG")
         p.stdin.close()
         p.wait()
-    model_path = f"{filename}.ply"
-    pc.save_ply(model_path)
+    model_path = f"{filename}.obj"
+    pc.save_obj(model_path)
     # model_path = None  # TODO
     return [video_path, model_path]
 

pulsar_clip.py

@@ -226,8 +226,8 @@ class PulsarCLIP(object):
             pass
     
     
-    def save_ply(self, fn):
-        utils.save_ply(self.get_points(), fn)
+    def save_obj(self, fn):
+        utils.save_obj(self.get_points(), fn)
 
 
 class DotDict(dict):

utils.py

@@ -120,7 +120,7 @@ def ico_at(xyz=np.array([0, 0, 0]), radius=1.0, i=0):
     return vert * radius + xyz, [[y + i for y in x] for x in idx]
 
 
-def save_ply(points, out_path):
+def save_obj(points, out_path):
     with torch.inference_mode():
         vert_pos, vert_col, vert_rad, vert_opa = (x.detach().cpu().numpy() for x in points)
       
@@ -129,24 +129,11 @@ def save_ply(points, out_path):
     for xyz, radius, (r, g, b), a in zip(vert_pos, vert_rad, vert_col, vert_opa):
         v, i = ico_at(xyz, radius, len(verts))
         for x, y, z in v:
-            verts.append((x, y, z, int(r * 255), int(g * 255), int(b * 255), int(a * 255)))
+            verts.append((x, y, z, r, g, b))  # int(r * 255), int(g * 255), int(b * 255), int(a * 255)))
         faces += i
 
     with open(out_path, "w") as out_file:
-        out_file.write("ply\n")
-        out_file.write("format ascii 1.0\n")
-        out_file.write(f"element vertex {len(verts)}\n")
-        out_file.write("property float x\n")
-        out_file.write("property float y\n")
-        out_file.write("property float z\n")
-        out_file.write("property uchar red\n")
-        out_file.write("property uchar green\n")
-        out_file.write("property uchar blue\n")
-        out_file.write("property uchar alpha\n")
-        out_file.write(f"element face {len(faces)}\n")
-        out_file.write("property list uchar int vertex_index\n")
-        out_file.write("end_header\n")
         for v in verts:
-            out_file.write(" ".join(map(str, v)) + "\n")
+            out_file.write("v " + " ".join(map(str, v)) + "\n")
         for f in faces:
-            out_file.write(" ".join(map(str, [len(f)] + f)) + "\n")
+            out_file.write("f " + " ".join(map(str, f)) + "\n")