YulianSa commited on
Commit
cfae272
·
1 Parent(s): af77592
Files changed (2) hide show
  1. app.py +2 -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 ~3min, please be patient.
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'>Github Repo</a>. Thanks! [![GitHub Stars](https://img.shields.io/github/stars/hyz317/StdGEN?style=social)](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:
 
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! [![GitHub Stars](https://img.shields.io/github/stars/hyz317/StdGEN?style=social)](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=120)
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