Update handcrafted_solution.py
Browse files- handcrafted_solution.py +1 -1
handcrafted_solution.py
CHANGED
@@ -212,7 +212,7 @@ def predict(entry, visualize=False) -> Tuple[np.ndarray, List[int]]:
|
|
212 |
gest_seg_np = np.array(gest_seg).astype(np.uint8)
|
213 |
# Metric3D
|
214 |
depth_np = np.array(depth) / 2.5 # 2.5 is the scale estimation coefficient
|
215 |
-
vertices, connections = get_vertices_and_edges_from_segmentation(gest_seg_np, edge_th =
|
216 |
if (len(vertices) < 2) or (len(connections) < 1):
|
217 |
print (f'Not enough vertices or connections in image {i}')
|
218 |
vert_edge_per_image[i] = np.empty((0, 2)), [], np.empty((0, 3))
|
|
|
212 |
gest_seg_np = np.array(gest_seg).astype(np.uint8)
|
213 |
# Metric3D
|
214 |
depth_np = np.array(depth) / 2.5 # 2.5 is the scale estimation coefficient
|
215 |
+
vertices, connections = get_vertices_and_edges_from_segmentation(gest_seg_np, edge_th = 5.)
|
216 |
if (len(vertices) < 2) or (len(connections) < 1):
|
217 |
print (f'Not enough vertices or connections in image {i}')
|
218 |
vert_edge_per_image[i] = np.empty((0, 2)), [], np.empty((0, 3))
|