Refactored mesh saving process
Browse files- predict.py +7 -10
predict.py
CHANGED
@@ -13,15 +13,6 @@ class Predictor(BasePredictor):
|
|
13 |
"MarcusLoren/MeshGPT-preview"
|
14 |
)
|
15 |
|
16 |
-
def save_as_obj(self, file_path):
|
17 |
-
v, f = igl.read_triangle_mesh(file_path)
|
18 |
-
v, f, _, _ = igl.remove_unreferenced(v, f)
|
19 |
-
c, _ = igl.orientable_patches(f)
|
20 |
-
f, _ = igl.orient_outward(v, f, c)
|
21 |
-
igl.write_triangle_mesh(file_path, v, f)
|
22 |
-
output_path = Path(tempfile.mkdtemp()) / file_path
|
23 |
-
return output_path
|
24 |
-
|
25 |
def predict(
|
26 |
self,
|
27 |
text: str = Input(description="Enter labels, separated by commas"),
|
@@ -62,7 +53,13 @@ class Predictor(BasePredictor):
|
|
62 |
(self.transformer.generate(texts=labels, temperature=num_temp))
|
63 |
)
|
64 |
file_name = "./mesh.obj"
|
65 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
66 |
file_path = self.save_as_obj(file_name)
|
67 |
return file_path
|
68 |
|
|
|
13 |
"MarcusLoren/MeshGPT-preview"
|
14 |
)
|
15 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
16 |
def predict(
|
17 |
self,
|
18 |
text: str = Input(description="Enter labels, separated by commas"),
|
|
|
53 |
(self.transformer.generate(texts=labels, temperature=num_temp))
|
54 |
)
|
55 |
file_name = "./mesh.obj"
|
56 |
+
file_path = Path(tempfile.mkdtemp()) / file_name
|
57 |
+
mesh_render.save_rendering(file_path, output)
|
58 |
+
v, f = igl.read_triangle_mesh(file_path)
|
59 |
+
v, f, _, _ = igl.remove_unreferenced(v, f)
|
60 |
+
c, _ = igl.orientable_patches(f)
|
61 |
+
f, _ = igl.orient_outward(v, f, c)
|
62 |
+
igl.write_triangle_mesh(file_path, v, f)
|
63 |
file_path = self.save_as_obj(file_name)
|
64 |
return file_path
|
65 |
|