run through training set to determine cluster numebr

handcrafted_solution.py

@@ -252,7 +252,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):
@@ -266,7 +266,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(XYZ, rgb, depth_np, gest_seg_np, K, R, t, dilate_r = 5):
   '''Project 3D sfm pointcloud to the image plane '''
@@ -295,7 +295,7 @@ def get_SfM_depth(XYZ, rgb, 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))
@@ -309,7 +309,7 @@ def get_SfM_depth(XYZ, rgb, 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')
@@ -784,8 +784,8 @@ def clean_PCD(XYZ, rgb):
   largest_blob_size = 0
   largest_blob = 0
   lowest_z = 0
-  print('uni label:', len(unique_labels))
-  if len(unique_labels) > 30:
+  #print('uni label:', len(unique_labels))
+  if len(unique_labels) > 40:
     return XYZ, rgb, lowest_z
 
   for k in unique_labels: