聂如
commited on
Commit
·
bbda53a
1
Parent(s):
c0763e9
Add design file
Browse files
app.py
CHANGED
@@ -317,6 +317,7 @@ def local_get_reconstructed_scene(inputfiles, min_conf_thr, cam_size):
|
|
317 |
pred_poses = torch.tensor(pred_poses)
|
318 |
# use knn to clean the point cloud
|
319 |
K = 10
|
|
|
320 |
points = torch.tensor(pts3d.reshape(1,-1,3)).cuda()
|
321 |
knn = knn_points(points, points, K=K)
|
322 |
dists = knn.dists
|
|
|
317 |
pred_poses = torch.tensor(pred_poses)
|
318 |
# use knn to clean the point cloud
|
319 |
K = 10
|
320 |
+
print('Cleaning point cloud with knn...')
|
321 |
points = torch.tensor(pts3d.reshape(1,-1,3)).cuda()
|
322 |
knn = knn_points(points, points, K=K)
|
323 |
dists = knn.dists
|