colin1842 commited on
Commit
1ce16d5
·
verified ·
1 Parent(s): 101a22c

Update handcrafted_solution.py

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