kcml commited on
Commit
a2b9eb7
·
1 Parent(s): 6b13967

[tune param] prune isolated only if pts are more than 35

Browse files
Files changed (1) hide show
  1. handcrafted_solution.py +1 -1
handcrafted_solution.py CHANGED
@@ -652,7 +652,7 @@ def predict(entry, visualize=False, prune_dist_thr=600, depth_scale=2.5, ) -> Tu
652
  #print(f'after merge, {len(all_3d_vertices)} 3d vertices and {len(connections_3d)} 3d connections')
653
  #all_3d_vertices_clean, connections_3d_clean = prune_not_connected(all_3d_vertices, connections_3d)
654
  all_3d_vertices, connections_3d = prune_tall(all_3d_vertices, connections_3d, 1000)
655
- if len(all_3d_vertices)>30:
656
  all_3d_vertices, connections_3d = prune_not_connected(all_3d_vertices, connections_3d)
657
  all_3d_vertices_clean, connections_3d_clean = prune_far(all_3d_vertices, connections_3d, prune_dist_thr=prune_dist_thr)
658
 
 
652
  #print(f'after merge, {len(all_3d_vertices)} 3d vertices and {len(connections_3d)} 3d connections')
653
  #all_3d_vertices_clean, connections_3d_clean = prune_not_connected(all_3d_vertices, connections_3d)
654
  all_3d_vertices, connections_3d = prune_tall(all_3d_vertices, connections_3d, 1000)
655
+ if len(all_3d_vertices)>35:
656
  all_3d_vertices, connections_3d = prune_not_connected(all_3d_vertices, connections_3d)
657
  all_3d_vertices_clean, connections_3d_clean = prune_far(all_3d_vertices, connections_3d, prune_dist_thr=prune_dist_thr)
658