bug fixed: type casting

handcrafted_solution.py

@@ -164,7 +164,7 @@ def get_smooth_uv_depth(vertices, depth, gest_seg_np, sfm_depth_np):
     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):
@@ -178,7 +178,6 @@ 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 '''
@@ -207,7 +206,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))
@@ -221,7 +220,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')
 
@@ -519,6 +518,7 @@ def delete_one_vert(vertices, vertices_3d, connections, vert_to_del):
   
   #print(f'del {len(conn_to_del)} connections')  
   connections = np.delete(connections, (conn_to_del), axis=0)
+  connections = connections.tolist()
   #print(vertices, vertices_3d, connections)
   if vertices:
     return vertices, vertices_3d, connections
@@ -599,7 +599,7 @@ def predict(entry, visualize=False, prune_dist_thr=600, depth_scale=2.5, ) -> Tu
             continue
                 
         #uv, depth_vert = get_uv_depth(vertices, depth_np)         
-        sfm_depth_np = get_SfM_depth(points3D, depth_np, gest_seg_np, K, R, t, 0)
+        sfm_depth_np = get_SfM_depth(points3D, depth_np, gest_seg_np, K, R, t, 10)
         uv, depth_vert = get_smooth_uv_depth(vertices, depth_np, gest_seg_np, sfm_depth_np)
         #uv, depth_vert = get_smooth_uv_depth(vertices, depth_np, gest_seg_np, None)        
 
@@ -616,9 +616,9 @@ def predict(entry, visualize=False, prune_dist_thr=600, depth_scale=2.5, ) -> Tu
     #all_3d_vertices, connections_3d = merge_vertices_3d(vert_edge_per_image, 3.0) # TODO: 3cm looks too small    
     all_3d_vertices, connections_3d = merge_vertices_3d(vert_edge_per_image, 30) 
     #print(f'after merge, {len(all_3d_vertices)} 3d vertices and {len(connections_3d)} 3d connections')
-    #all_3d_vertices_clean, connections_3d_clean  = prune_not_connected(all_3d_vertices, connections_3d)
-    #all_3d_vertices_clean, connections_3d_clean = prune_far(all_3d_vertices, connections_3d, prune_dist_thr=prune_dist_thr)
-    all_3d_vertices_clean, connections_3d_clean  = all_3d_vertices, connections_3d # don't prune -> cost:2.0
+    #all_3d_vertices_clean, connections_3d_clean  = prune_not_connected(all_3d_vertices, connections_3d)    
+    all_3d_vertices_clean, connections_3d_clean = prune_far(all_3d_vertices, connections_3d, prune_dist_thr=prune_dist_thr)
+    #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:
         print (f'Not enough vertices or connections in the 3D vertices')