pengHTYX commited on
Commit
378ba83
1 Parent(s): e67791b
app.py CHANGED
@@ -222,7 +222,7 @@ def run_pipeline(pipeline, cfg, single_image, guidance_scale, steps, seed, crop_
222
  num_views = 6
223
  if write_image:
224
  VIEWS = ['front', 'front_right', 'right', 'back', 'left', 'front_left']
225
- cur_dir = os.path.join("./mv_res", f"cropsize-{int(crop_size)}-cfg{guidance_scale:.1f}")
226
 
227
  scene = 'scene'+datetime.now().strftime('@%Y%m%d-%H%M%S')
228
  scene_dir = os.path.join(cur_dir, scene)
 
222
  num_views = 6
223
  if write_image:
224
  VIEWS = ['front', 'front_right', 'right', 'back', 'left', 'front_left']
225
+ cur_dir = os.path.join(cfg.save_dir, f"cropsize-{int(crop_size)}-cfg{guidance_scale:.1f}")
226
 
227
  scene = 'scene'+datetime.now().strftime('@%Y%m%d-%H%M%S')
228
  scene_dir = os.path.join(cur_dir, scene)
configs/test_unclip-512-6view.yaml CHANGED
@@ -13,7 +13,7 @@ validation_dataset:
13
  crop_size: 420
14
 
15
  pred_type: 'joint'
16
- save_dir: 'mv_results'
17
  save_mode: 'rgba' # 'concat', 'rgba', 'rgb'
18
  seed: 42
19
  validation_batch_size: 1
 
13
  crop_size: 420
14
 
15
  pred_type: 'joint'
16
+ save_dir: 'mv_res'
17
  save_mode: 'rgba' # 'concat', 'rgba', 'rgb'
18
  seed: 42
19
  validation_batch_size: 1
sam_pt/sam_vit_h_4b8939.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a7bf3b02f3ebf1267aba913ff637d9a2d5c33d3173bb679e46d9f338c26f262e
3
+ size 2564550879