xinjie.wang commited on
Commit
66e2971
·
1 Parent(s): 3075458
Files changed (2) hide show
  1. common.py +3 -8
  2. embodied_gen/scripts/imageto3d.py +3 -8
common.py CHANGED
@@ -506,13 +506,8 @@ def extract_3d_representations_v2(
506
  render_gs_api(
507
  input_gs=aligned_gs_path,
508
  output_path=color_path,
509
- elevation=[20, -10],
510
- )
511
- color_path2 = os.path.join(user_dir, "color2.png")
512
- render_gs_api(
513
- input_gs=aligned_gs_path,
514
- output_path=color_path2,
515
- elevation=[60, -50],
516
  )
517
 
518
  mesh = trimesh.Trimesh(
@@ -528,7 +523,7 @@ def extract_3d_representations_v2(
528
  mesh = backproject_api(
529
  delight_model=DELIGHT,
530
  imagesr_model=IMAGESR_MODEL,
531
- color_path=[color_path, color_path2],
532
  mesh_path=mesh_obj_path,
533
  output_path=mesh_obj_path,
534
  skip_fix_mesh=False,
 
506
  render_gs_api(
507
  input_gs=aligned_gs_path,
508
  output_path=color_path,
509
+ elevation=[20, -10, 60, -50],
510
+ num_images=12,
 
 
 
 
 
511
  )
512
 
513
  mesh = trimesh.Trimesh(
 
523
  mesh = backproject_api(
524
  delight_model=DELIGHT,
525
  imagesr_model=IMAGESR_MODEL,
526
+ color_path=color_path,
527
  mesh_path=mesh_obj_path,
528
  output_path=mesh_obj_path,
529
  skip_fix_mesh=False,
embodied_gen/scripts/imageto3d.py CHANGED
@@ -215,13 +215,8 @@ def entrypoint(**kwargs):
215
  render_gs_api(
216
  input_gs=aligned_gs_path,
217
  output_path=color_path,
218
- elevation=[20, -10],
219
- )
220
- color_path2 = os.path.join(output_root, "color2.png")
221
- render_gs_api(
222
- input_gs=aligned_gs_path,
223
- output_path=color_path2,
224
- elevation=[60, -50],
225
  )
226
 
227
  geo_flag, geo_result = GEO_CHECKER(
@@ -254,7 +249,7 @@ def entrypoint(**kwargs):
254
  mesh = backproject_api(
255
  delight_model=DELIGHT,
256
  imagesr_model=IMAGESR_MODEL,
257
- color_path=[color_path, color_path2],
258
  mesh_path=mesh_obj_path,
259
  output_path=mesh_obj_path,
260
  skip_fix_mesh=False,
 
215
  render_gs_api(
216
  input_gs=aligned_gs_path,
217
  output_path=color_path,
218
+ elevation=[20, -10, 60, -50],
219
+ num_images=12,
 
 
 
 
 
220
  )
221
 
222
  geo_flag, geo_result = GEO_CHECKER(
 
249
  mesh = backproject_api(
250
  delight_model=DELIGHT,
251
  imagesr_model=IMAGESR_MODEL,
252
+ color_path=color_path,
253
  mesh_path=mesh_obj_path,
254
  output_path=mesh_obj_path,
255
  skip_fix_mesh=False,