[tune param] prune isolated only if pts are more than 35
Browse files- 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)>
|
| 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 |
|