mit-reality-hack-img-3d

Running on Zero

App Files Files Community

Huiwenshi commited on 10 days ago

Commit

002ae29

verified ·

1 Parent(s): 2922df3

Upload folder using huggingface_hub

Browse files

Files changed (8) hide show

.gitattributes +1 -0
gradio_cache/5/input.png +0 -0
gradio_cache/5/rembg.png +0 -0
gradio_cache/5/textured_mesh.glb +3 -0
gradio_cache/5/textured_mesh.html +40 -0
gradio_cache/5/white_mesh.glb +0 -0
gradio_cache/5/white_mesh.html +57 -0
hg_app.py +1 -40

.gitattributes CHANGED Viewed

@@ -41,3 +41,4 @@ assets/images/teaser.jpg filter=lfs diff=lfs merge=lfs -text
 gradio_cache/0/textured_mesh.glb filter=lfs diff=lfs merge=lfs -text
 gradio_cache/3/textured_mesh.glb filter=lfs diff=lfs merge=lfs -text
 gradio_cache/4/textured_mesh.glb filter=lfs diff=lfs merge=lfs -text

 gradio_cache/0/textured_mesh.glb filter=lfs diff=lfs merge=lfs -text
 gradio_cache/3/textured_mesh.glb filter=lfs diff=lfs merge=lfs -text
 gradio_cache/4/textured_mesh.glb filter=lfs diff=lfs merge=lfs -text
+gradio_cache/5/textured_mesh.glb filter=lfs diff=lfs merge=lfs -text

gradio_cache/5/input.png ADDED Viewed

gradio_cache/5/rembg.png ADDED Viewed

gradio_cache/5/textured_mesh.glb ADDED Viewed

	@@ -0,0 +1,3 @@

+version https://git-lfs.github.com/spec/v1
+oid sha256:acc1b80df3fa7ab7b865660a34b6f57db16c1e9e1806a8cbb766d7a75bab125b
+size 2230664

gradio_cache/5/textured_mesh.html ADDED Viewed

	@@ -0,0 +1,40 @@

+<!DOCTYPE html>
+<html>
+<head>
+    <!-- Import the component -->
+    <script src="https://ajax.googleapis.com/ajax/libs/model-viewer/3.1.1/model-viewer.min.js" type="module"></script>
+    <style>
+        body {
+            margin: 0;
+            font-family: Arial, sans-serif;
+        }
+        .centered-container {
+            display: flex;
+            justify-content: center;
+            align-items: center;
+            border-radius: 8px;
+            border-color: #e5e7eb;
+            border-style: solid;
+            border-width: 1px;
+        }
+    </style>
+</head>
+<body>
+<div class="centered-container">
+            <div class="column is-mobile is-centered">
+                <model-viewer style="height: 586px; width: 700px;" rotation-per-second="10deg" id="modelViewer"
+                    src="./textured_mesh.glb/" disable-tap
+                    environment-image="neutral" auto-rotate camera-target="0m 0m 0m" orientation="0deg 0deg 170deg" shadow-intensity=".9"
+                    ar auto-rotate camera-controls>
+                </model-viewer>
+            </div>
+</div>
+</body>
+</html>

gradio_cache/5/white_mesh.glb ADDED Viewed

Binary file (721 kB). View file

gradio_cache/5/white_mesh.html ADDED Viewed

	@@ -0,0 +1,57 @@

+<!DOCTYPE html>
+<html>
+<head>
+    <!-- Import the component -->
+    <script src="https://ajax.googleapis.com/ajax/libs/model-viewer/3.1.1/model-viewer.min.js" type="module"></script>
+    <script>
+        document.addEventListener('DOMContentLoaded', () => {
+            const modelViewers = document.querySelectorAll('model-viewer');
+            modelViewers.forEach(modelViewer => {
+                modelViewer.addEventListener('load', (event) => {
+                    const [material] = modelViewer.model.materials;
+                    let color = [43, 44, 46, 255];
+                    color = color.map(x => x / 255);
+                    material.pbrMetallicRoughness.setMetallicFactor(0.1); // 完全金属
+                    material.pbrMetallicRoughness.setRoughnessFactor(0.7); // 低粗糙度
+                    material.pbrMetallicRoughness.setBaseColorFactor(color);  // CornflowerBlue in RGB
+                });
+            });
+        });
+    </script>
+    <style>
+        body {
+            margin: 0;
+            font-family: Arial, sans-serif;
+        }
+        .centered-container {
+            display: flex;
+            justify-content: center;
+            align-items: center;
+            border-radius: 8px;
+            border-color: #e5e7eb;
+            border-style: solid;
+            border-width: 1px;
+        }
+    </style>
+</head>
+<body>
+<div class="centered-container">
+            <div class="column is-mobile is-centered">
+                <model-viewer style="height: 586px; width: 700px;" rotation-per-second="10deg" id="modelViewer"
+                    src="./white_mesh.glb/" disable-tap
+                    environment-image="neutral" auto-rotate camera-target="0m 0m 0m" orientation="0deg 0deg 170deg" shadow-intensity=".9"
+                    ar auto-rotate camera-controls>
+                </model-viewer>
+            </div>
+</div>
+</body>
+</html>

hg_app.py CHANGED Viewed

@@ -6,49 +6,10 @@ if True:
     import sys
     import shlex
-    def install_package(package_path):
-        # 确保 package_path 是绝对路径
-        package_path = os.path.abspath(package_path)
-        # 设置环境变量
-        env = os.environ.copy()  # 复制当前环境变量
-        env['CUDA_HOME'] = '/usr/local/cuda'
-        env['FORCE_CUDA'] = '1'
-        env['TORCH_CUDA_ARCH_LIST'] = '8.0;8.6;8.9;9.0'
-        # 使用 subprocess 调用 setup.py
-        try:
-            subprocess.check_call([sys.executable, os.path.join(package_path, 'setup.py'), 'install'], env=env)
-            print(f"Package installed from {package_path}")
-        except subprocess.CalledProcessError as e:
-            print(f"Failed to install package from {package_path}: {e}")
-    def install_cuda_toolkit():
-        # CUDA_TOOLKIT_URL = "https://developer.download.nvidia.com/compute/cuda/11.8.0/local_installers/cuda_11.8.0_520.61.05_linux.run"
-        CUDA_TOOLKIT_URL = "https://developer.download.nvidia.com/compute/cuda/12.2.0/local_installers/cuda_12.2.0_535.54.03_linux.run"
-        CUDA_TOOLKIT_FILE = "/tmp/%s" % os.path.basename(CUDA_TOOLKIT_URL)
-        subprocess.call(["wget", "-q", CUDA_TOOLKIT_URL, "-O", CUDA_TOOLKIT_FILE])
-        subprocess.call(["chmod", "+x", CUDA_TOOLKIT_FILE])
-        subprocess.call([CUDA_TOOLKIT_FILE, "--silent", "--toolkit"])
-        os.environ["CUDA_HOME"] = "/usr/local/cuda"
-        os.environ["PATH"] = "%s/bin:%s" % (os.environ["CUDA_HOME"], os.environ["PATH"])
-        os.environ["LD_LIBRARY_PATH"] = "%s/lib:%s" % (
-            os.environ["CUDA_HOME"],
-            "" if "LD_LIBRARY_PATH" not in os.environ else os.environ["LD_LIBRARY_PATH"],
-        )
-        # Fix: arch_list[-1] += '+PTX'; IndexError: list index out of range
-        os.environ["TORCH_CUDA_ARCH_LIST"] = "8.0;8.6"
-    # install_cuda_toolkit()
     print("cd /home/user/app/hy3dgen/texgen/differentiable_renderer/ && bash compile_mesh_painter.sh")
     os.system("cd /home/user/app/hy3dgen/texgen/differentiable_renderer/ && bash compile_mesh_painter.sh")
-    # print("cd /home/user/app/hy3dgen/texgen/custom_rasterizer && python3 -m pip install .")
-    # os.system("cd /home/user/app/hy3dgen/texgen/custom_rasterizer && python3 -m pip install .")
     print('install custom')
-    # install_package("/home/user/app/hy3dgen/texgen/custom_rasterizer")
     subprocess.run(shlex.split("pip install . --no-build-isolation"), cwd="/home/user/app/hy3dgen/texgen/custom_rasterizer/", check=True)
-    # os.system("cd /home/user/app/hy3dgen/texgen/custom_rasterizer && CUDA_HOME=/usr/local/cuda FORCE_CUDA=1 TORCH_CUDA_ARCH_LIST='8.0;8.6;8.9;9.0' python setup.py install")
     IP = "0.0.0.0"
     PORT = 7860
@@ -457,6 +418,6 @@ if __name__ == '__main__':
     app.mount("/static", StaticFiles(directory=static_dir), name="static")
     demo = build_app()
-    demo.queue(max_size=2)
     app = gr.mount_gradio_app(app, demo, path="/")
     uvicorn.run(app, host=IP, port=PORT)

     import sys
     import shlex
     print("cd /home/user/app/hy3dgen/texgen/differentiable_renderer/ && bash compile_mesh_painter.sh")
     os.system("cd /home/user/app/hy3dgen/texgen/differentiable_renderer/ && bash compile_mesh_painter.sh")
     print('install custom')
     subprocess.run(shlex.split("pip install . --no-build-isolation"), cwd="/home/user/app/hy3dgen/texgen/custom_rasterizer/", check=True)
     IP = "0.0.0.0"
     PORT = 7860
     app.mount("/static", StaticFiles(directory=static_dir), name="static")
     demo = build_app()
+    demo.queue(max_size=1)
     app = gr.mount_gradio_app(app, demo, path="/")
     uvicorn.run(app, host=IP, port=PORT)