Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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))
|