Spaces:
Running
on
Zero
Running
on
Zero
daidedou
commited on
Commit
·
259b118
1
Parent(s):
20ce39e
Remove pykeops for space ZeroGPU demo.
Browse files
app.py
CHANGED
|
@@ -210,7 +210,8 @@ def run_clicked(mesh1_path, mesh2_path, yaml_path, lambda_val, zoomout_val, time
|
|
| 210 |
C12_new, p2p, p2p_init, _, loss_save = matcher.optimize(datadicts.shape_dict, datadicts.target_dict, target_normals)
|
| 211 |
evecs1, evecs2 = datadicts.shape_dict["evecs"], datadicts.target_dict["evecs"]
|
| 212 |
evecs_2trans = evecs2.t() @ torch.diag(datadicts.target_dict["mass"])
|
| 213 |
-
|
|
|
|
| 214 |
p2p_zo, _ = extract_p2p_torch_fmap(C12_end_zo, datadicts.shape_dict["evecs"], datadicts.target_dict["evecs"])
|
| 215 |
return build_outputs(datadicts.shape_surf, datadicts.target_surf, datadicts.cmap1, p2p_zo, tag="run")
|
| 216 |
|
|
|
|
| 210 |
C12_new, p2p, p2p_init, _, loss_save = matcher.optimize(datadicts.shape_dict, datadicts.target_dict, target_normals)
|
| 211 |
evecs1, evecs2 = datadicts.shape_dict["evecs"], datadicts.target_dict["evecs"]
|
| 212 |
evecs_2trans = evecs2.t() @ torch.diag(datadicts.target_dict["mass"])
|
| 213 |
+
with torch.no_grad():
|
| 214 |
+
C12_end_zo = torch_zoomout(evecs1, evecs2, evecs_2trans, C12_new.squeeze()[:15, :15], matcher.cfg.sds_conf.zoomout)
|
| 215 |
p2p_zo, _ = extract_p2p_torch_fmap(C12_end_zo, datadicts.shape_dict["evecs"], datadicts.target_dict["evecs"])
|
| 216 |
return build_outputs(datadicts.shape_surf, datadicts.target_surf, datadicts.cmap1, p2p_zo, tag="run")
|
| 217 |
|