Spaces:
Sleeping
Sleeping
FrozenBurning
commited on
Commit
·
e1329d4
1
Parent(s):
a4912eb
Update app.py
Browse files
app.py
CHANGED
@@ -137,7 +137,7 @@ def process(input_cond, input_num_steps, input_seed=42, input_cfg=6.0):
|
|
137 |
recon_feat_param[:, 1:, ...] = (recon_feat_param[:, 1:, ...] + 1) / 2.
|
138 |
recon_feat_param = recon_feat_param.reshape(inf_bs, config.model.num_prims, -1)
|
139 |
recon_param = torch.concat([recon_srt_param, recon_feat_param], dim=-1)
|
140 |
-
visualize_video_primvolume(config.output_dir, batch, recon_param,
|
141 |
prim_params = {'srt_param': recon_srt_param[0].detach().cpu(), 'feat_param': recon_feat_param[0].detach().cpu()}
|
142 |
torch.save({'model_state_dict': prim_params}, "{}/denoised.pt".format(config.output_dir))
|
143 |
|
@@ -200,7 +200,7 @@ with block:
|
|
200 |
# gen button
|
201 |
button_gen = gr.Button(value="Generate", interactive=False)
|
202 |
with gr.Row():
|
203 |
-
input_mc_resolution = gr.Radio(choices=[
|
204 |
input_remesh = gr.Radio(choices=["No", "Yes"], label="Remesh", value="No", info="Remesh or not?")
|
205 |
export_glb_btn = gr.Button(value="Export GLB", interactive=False)
|
206 |
|
|
|
137 |
recon_feat_param[:, 1:, ...] = (recon_feat_param[:, 1:, ...] + 1) / 2.
|
138 |
recon_feat_param = recon_feat_param.reshape(inf_bs, config.model.num_prims, -1)
|
139 |
recon_param = torch.concat([recon_srt_param, recon_feat_param], dim=-1)
|
140 |
+
visualize_video_primvolume(config.output_dir, batch, recon_param, 15, rm, device)
|
141 |
prim_params = {'srt_param': recon_srt_param[0].detach().cpu(), 'feat_param': recon_feat_param[0].detach().cpu()}
|
142 |
torch.save({'model_state_dict': prim_params}, "{}/denoised.pt".format(config.output_dir))
|
143 |
|
|
|
200 |
# gen button
|
201 |
button_gen = gr.Button(value="Generate", interactive=False)
|
202 |
with gr.Row():
|
203 |
+
input_mc_resolution = gr.Radio(choices=[64, 128, 256], label="MC Resolution", value=128, info="Cube resolution for mesh extraction")
|
204 |
input_remesh = gr.Radio(choices=["No", "Yes"], label="Remesh", value="No", info="Remesh or not?")
|
205 |
export_glb_btn = gr.Button(value="Export GLB", interactive=False)
|
206 |
|