daidedou commited on
Commit
259b118
·
1 Parent(s): 20ce39e

Remove pykeops for space ZeroGPU demo.

Browse files
Files changed (1) hide show
  1. app.py +2 -1
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
- C12_end_zo = torch_zoomout(evecs1, evecs2, evecs_2trans, C12_new.squeeze()[:15, :15], 150)# matcher.cfg.sds_conf.zoomout)
 
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