Spaces:
Running
on
L40S
Running
on
L40S
update
Browse files- app.py +2 -2
- infer_api.py +1 -2
app.py
CHANGED
@@ -48,7 +48,7 @@ This is official demo for our CVPR 2025 paper <a href="">StdGEN: Semantic-Decomp
|
|
48 |
Code: <a href='https://github.com/hyz317/StdGEN' target='_blank'>GitHub</a>. Paper: <a href='https://arxiv.org/abs/2411.05738' target='_blank'>ArXiv</a>.
|
49 |
|
50 |
❗️❗️❗️**Important Notes:**
|
51 |
-
1. Refinement stage takes about ~
|
52 |
|
53 |
2. You can upload any reference image (with or without background), A-pose images are also supported (white bkg required). If the image has an alpha channel (transparency), background segmentation will be automatically performed. Alternatively, you can pre-segment the background using other tools and upload the result directly.
|
54 |
|
@@ -58,7 +58,7 @@ Code: <a href='https://github.com/hyz317/StdGEN' target='_blank'>GitHub</a>. Pap
|
|
58 |
'''
|
59 |
|
60 |
_CITE_ = r"""
|
61 |
-
If StdGEN is helpful, please help to ⭐ the <a href='https://github.com/hyz317/StdGEN' target='_blank'>
|
62 |
---
|
63 |
📝 **Citation**
|
64 |
If you find our work useful for your research or applications, please cite using this bibtex:
|
|
|
48 |
Code: <a href='https://github.com/hyz317/StdGEN' target='_blank'>GitHub</a>. Paper: <a href='https://arxiv.org/abs/2411.05738' target='_blank'>ArXiv</a>.
|
49 |
|
50 |
❗️❗️❗️**Important Notes:**
|
51 |
+
1. Refinement stage takes about ~3.5min, and the mesh result may possibly delayed due to the server load, please wait patiently.
|
52 |
|
53 |
2. You can upload any reference image (with or without background), A-pose images are also supported (white bkg required). If the image has an alpha channel (transparency), background segmentation will be automatically performed. Alternatively, you can pre-segment the background using other tools and upload the result directly.
|
54 |
|
|
|
58 |
'''
|
59 |
|
60 |
_CITE_ = r"""
|
61 |
+
If StdGEN is helpful, please help to ⭐ the <a href='https://github.com/hyz317/StdGEN' target='_blank'>GitHub Repo</a>. Thanks! [](https://github.com/hyz317/StdGEN)
|
62 |
---
|
63 |
📝 **Citation**
|
64 |
If you find our work useful for your research or applications, please cite using this bibtex:
|
infer_api.py
CHANGED
@@ -427,7 +427,6 @@ def srgb_to_linear(c_srgb):
|
|
427 |
return c_linear.clip(0, 1.)
|
428 |
|
429 |
|
430 |
-
@spaces.GPU
|
431 |
def save_py3dmesh_with_trimesh_fast(meshes: Meshes, save_glb_path, apply_sRGB_to_LinearRGB=True):
|
432 |
# convert from pytorch3d meshes to trimesh mesh
|
433 |
vertices = meshes.verts_packed().cpu().float().numpy()
|
@@ -555,7 +554,7 @@ infer_refine_generator = SamAutomaticMaskGenerator(
|
|
555 |
)
|
556 |
infer_refine_outside_ratio = 0.20
|
557 |
|
558 |
-
@spaces.GPU(duration=
|
559 |
def infer_refine(meshes, imgs):
|
560 |
fixed_v, fixed_f, fixed_t = None, None, None
|
561 |
flow_vert, flow_vector = None, None
|
|
|
427 |
return c_linear.clip(0, 1.)
|
428 |
|
429 |
|
|
|
430 |
def save_py3dmesh_with_trimesh_fast(meshes: Meshes, save_glb_path, apply_sRGB_to_LinearRGB=True):
|
431 |
# convert from pytorch3d meshes to trimesh mesh
|
432 |
vertices = meshes.verts_packed().cpu().float().numpy()
|
|
|
554 |
)
|
555 |
infer_refine_outside_ratio = 0.20
|
556 |
|
557 |
+
@spaces.GPU(duration=150)
|
558 |
def infer_refine(meshes, imgs):
|
559 |
fixed_v, fixed_f, fixed_t = None, None, None
|
560 |
flow_vert, flow_vector = None, None
|