winfred2027 commited on
Commit
3033bd9
·
verified ·
1 Parent(s): 01c4a37

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -2
app.py CHANGED
@@ -119,8 +119,6 @@ def retrieval_pc(load_data, k, sim_th, filter_fn):
119
  col2 = utils.render_pc(pc)
120
  ref_dev = next(model_retrieval.parameters()).device
121
  enc = model_retrieval(torch.tensor(pc[:, [0, 2, 1, 3, 4, 5]].T[None], device=ref_dev))
122
- if model_name == "pb-sn-M":
123
- enc = pc_adapter(enc)
124
  sim = torch.matmul(torch.nn.functional.normalize(lvis.feats, dim=-1), torch.nn.functional.normalize(enc.cpu(), dim=-1).squeeze())
125
  argsort = torch.argsort(sim, descending=True)
126
  pred = OrderedDict((lvis.categories[i], sim[i]) for i in argsort if i < len(lvis.categories))
 
119
  col2 = utils.render_pc(pc)
120
  ref_dev = next(model_retrieval.parameters()).device
121
  enc = model_retrieval(torch.tensor(pc[:, [0, 2, 1, 3, 4, 5]].T[None], device=ref_dev))
 
 
122
  sim = torch.matmul(torch.nn.functional.normalize(lvis.feats, dim=-1), torch.nn.functional.normalize(enc.cpu(), dim=-1).squeeze())
123
  argsort = torch.argsort(sim, descending=True)
124
  pred = OrderedDict((lvis.categories[i], sim[i]) for i in argsort if i < len(lvis.categories))