dummy conn

handcrafted_solution.py

@@ -180,23 +180,18 @@ def get_smooth_uv_depth(vertices, depth, gest_seg_np, sfm_depth_np, r=5):
       '''Search on sfm depth first. Search on depthmap only if no sfm depth 
       exists at all in the local region.
       '''
-      r = 5
-      yslice = slice(max(0, y - r), min(H, y + r))
-      xslice = slice(max(0, x - r), min(W, x + r))      
-      local_area = sfm_depth_np[yslice, xslice]
-      reduced_local_area = local_area[local_area!=0]
       PRINT = False 
-      #if np.isclose(x, 2574.81093605, atol=10) and np.isclose(y, 1063.7265987, atol=10):
-      #  PRINT = True
-      
-      while (reduced_local_area.size == 0) and r<200:
-        #print('r=', r)
-        r*=2
+      r_choices = [75]
+      for r in r_choices:
         yslice = slice(max(0, y - r), min(H, y + r))
         xslice = slice(max(0, x - r), min(W, x + r))      
         local_area = sfm_depth_np[yslice, xslice]
-        reduced_local_area = local_area[local_area!=0]            
-
+        reduced_local_area = local_area[local_area!=0]
+        #if np.isclose(x, 2574.81093605, atol=10) and np.isclose(y, 1063.7265987, atol=10):
+        #  PRINT = True
+        if reduced_local_area.size > 0:
+          break              
+      #print('r=', r)
       if reduced_local_area.size > 0:
         #print('use sfm')
         if PRINT: print(reduced_local_area)
@@ -256,7 +251,7 @@ def get_smooth_uv_depth(vertices, depth, gest_seg_np, sfm_depth_np, r=5):
     vertex_depth = np.array(vertex_depth)
     return uv, vertex_depth
 
-'''
+
 from numba import njit, prange
 @njit(parallel=True)
 def fill_range(u, v, z, dilate_r, c, sfm_depth_np, sfm_color_np, H, W):
@@ -270,7 +265,7 @@ def fill_range(u, v, z, dilate_r, c, sfm_depth_np, sfm_color_np, H, W):
             if DUMP_IMG: 
               sfm_color_np[j, i] = c
   return sfm_depth_np, sfm_color_np
-'''
+''''''
 
 def get_SfM_depth(points3D, depth_np, gest_seg_np, K, R, t, dilate_r = 5):
   '''Project 3D sfm pointcloud to the image plane '''
@@ -299,7 +294,7 @@ def get_SfM_depth(points3D, depth_np, gest_seg_np, K, R, t, dilate_r = 5):
   #checked = 0  
   #print('dim of us uv zs rgb:', len(us), len(vs), len(zs), len(rgb))
   for u,v,z,c in zip(us,vs,zs, rgb):
-    '''
+    
     sfm_depth_np, sfm_color_np = fill_range(u, v, z, dilate_r, c, sfm_depth_np, sfm_color_np, H, W)
     '''
     i_range = range(max(0, u - dilate_r), min(W, u + dilate_r))
@@ -313,7 +308,7 @@ def get_SfM_depth(points3D, depth_np, gest_seg_np, K, R, t, dilate_r = 5):
             sfm_depth_np[j, i] = z
             if DUMP_IMG: 
               sfm_color_np[j, i] = c
-    
+    '''
          
               
   #print(f'checked {checked} pts')
@@ -424,7 +419,7 @@ def get_vertices_and_edges_from_two_segmentations(ade_seg_np, gest_seg_np, edge_
         apex_pts.append(v['xy'])
         apex_pts_idxs.append(j)
     apex_pts = np.array(apex_pts)
-            
+    '''       
     # Ridge connects two apex points
     def Ridge_connects_two_apex_points(gest_seg_np, color_th, apex_pts, edge_th):
       conn = []
@@ -511,7 +506,7 @@ def get_vertices_and_edges_from_two_segmentations(ade_seg_np, gest_seg_np, edge_
 
       return conn, mask
     connections, line_img = classifyPairs(apex_pts, apex_pts_idxs, gest_seg_np, apex_mask, eave_end_mask)
-    '''
+    
     #print(f'{len(vertices)} vertices: {vertices}')
     #print(len(connections), ' connections: ', connections)
     if DUMP_IMG: 
@@ -808,6 +803,8 @@ def predict(entry, visualize=False, prune_dist_thr=600, depth_scale=2.5, ) -> Tu
       all_3d_vertices_clean, connections_3d_clean = prune_far(all_3d_vertices, connections_3d, prune_dist_thr=prune_dist_thr)
     else:
       all_3d_vertices_clean, connections_3d_clean = all_3d_vertices, connections_3d
+
+    connections_3d_clean = [(0, 1)]
     #all_3d_vertices_clean, connections_3d_clean  = all_3d_vertices, connections_3d # don't prune -> cost:2.0
     #print(f'after pruning, {len(all_3d_vertices_clean)} 3d clean vertices and {len(connections_3d_clean)} 3d clean connections')    
     if (len(all_3d_vertices_clean) < 2) or len(connections_3d_clean) < 1: