Spaces:
Runtime error
Runtime error
HongFangzhou
commited on
Commit
β’
f5d98e9
1
Parent(s):
bc2085d
add source codes
Browse files
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
|