Siromanec commited on
Commit
2cbdf26
·
verified ·
1 Parent(s): 88a232d

Update handcrafted_solution.py

Browse files
Files changed (1) hide show
  1. 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=50.)
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))