initial commit

.gitattributes

@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+*.obj filter=lfs diff=lfs merge=lfs -text

.gitignore

@@ -0,0 +1,4 @@
+__pycache__
+
+gradio_cached_examples/
+venv/

app.py

@@ -0,0 +1,38 @@
+import gradio as gr
+import trimesh
+from instant_texture import Converter
+
+converter = Converter()
+
+
+def extract_texture(mesh_path: str):
+    mesh = trimesh.load(mesh_path)
+    mesh = next(iter(mesh.geometry.values()))
+    texture = mesh.visual.material.baseColorTexture
+    return texture
+
+
+def convert(input_mesh_path: str):
+    if not input_mesh_path.endswith(".obj"):
+        raise ValueError("Only .obj files are supported")
+    output_path = "/tmp/output.glb"
+    converter.convert(input_mesh_path, output_path)
+    texture = extract_texture(output_path)
+    return output_path, texture
+
+
+demo = gr.Interface(
+    convert,
+    title="Instant Texture",
+    description="Convert a vertex-colored mesh (.obj) to a uv-mapped, textured mesh (.glb).",
+    inputs=gr.Model3D(label="Vertex-colored mesh (.obj)"),
+    outputs=[
+        gr.Model3D(label="Output uv-mapped, textured mesh (.glb)"),
+        gr.Image(label="Output texture"),
+    ],
+    examples=["examples/chair.obj"],
+    cache_examples=True,
+)
+
+if __name__ == "__main__":
+    demo.queue().launch()

examples/chair.obj

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:2d348aa68e6d3fc2b1359c3c60b227dc47484421cd0ce0bde81f386a4ec94ffa
+size 4453602

requirements.txt

@@ -0,0 +1 @@
+instant-texture @ git+https://github.com/dylanebert/InstantTexture