slow loops

handcrafted_solution.py

@@ -164,6 +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):
@@ -177,6 +178,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 '''
@@ -202,8 +204,8 @@ def get_SfM_depth(points3D, depth_np, gest_seg_np, K, R, t, dilate_r = 5):
   #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)
-    '''
+    #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))
     j_range = range(max(0, v - dilate_r), min(H, v + dilate_r)) 
     for i in i_range:
@@ -215,7 +217,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')
 

script.py

@@ -103,6 +103,7 @@ def convert_entry_to_human_readable(entry):
 ### The part below is used to define and test your solution.
 
 ### The function below is useful for installing additional python wheels.  
+'''
 import subprocess
 def my_install_package_from_local_file(package_name, folder='packages'):
      """
@@ -121,7 +122,7 @@ def my_install_package_from_local_file(package_name, folder='packages'):
          print(f"Failed to install package from {pth}. Error: {e}")
 
 my_install_package_from_local_file('numba')
-
+'''
 from pathlib import Path
 def save_submission(submission, path):
     """