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