HongFangzhou commited on
Commit
f5d98e9
β€’
1 Parent(s): bc2085d

add source codes

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -51,7 +51,7 @@ def add_text(rgb, caption):
51
  cv2.putText(rgb, bci, (gap, gap*(i+1)), font, fontScale, color, thickness, cv2.LINE_AA)
52
  return rgb
53
 
54
- config = "configs/default.yaml"
55
  local_ckpt = "checkpoints/3dtopia_diffusion_state_dict.ckpt"
56
  if os.path.exists(local_ckpt):
57
  ckpt = local_ckpt
@@ -76,7 +76,7 @@ img_size = configs.model.params.unet_config.params.image_size
76
  channels = configs.model.params.unet_config.params.in_channels
77
  shape = [channels, img_size, img_size * 3]
78
 
79
- pose_folder = 'assets/sample_data/pose'
80
  poses_fname = sorted([os.path.join(pose_folder, f) for f in os.listdir(pose_folder)])
81
  batch_rays_list = []
82
  H = 128
 
51
  cv2.putText(rgb, bci, (gap, gap*(i+1)), font, fontScale, color, thickness, cv2.LINE_AA)
52
  return rgb
53
 
54
+ config = "3DTopia/configs/default.yaml"
55
  local_ckpt = "checkpoints/3dtopia_diffusion_state_dict.ckpt"
56
  if os.path.exists(local_ckpt):
57
  ckpt = local_ckpt
 
76
  channels = configs.model.params.unet_config.params.in_channels
77
  shape = [channels, img_size, img_size * 3]
78
 
79
+ pose_folder = '3DTopia/assets/sample_data/pose'
80
  poses_fname = sorted([os.path.join(pose_folder, f) for f in os.listdir(pose_folder)])
81
  batch_rays_list = []
82
  H = 128