Hugging Face's logo

kcml
/
handcrafted_baseline_submission

Model card Files Community
kcml commited on
Commit
5d45331
·
1 Parent(s): 4efc279

[test]No numba

Browse files
Files changed (1) hide show
  1. handcrafted_solution.py +3 -3
handcrafted_solution.py CHANGED
@@ -599,9 +599,9 @@ def predict(entry, visualize=False, prune_dist_thr=600, depth_scale=2.5, ) -> Tu
599
  continue
600
 
601
  #uv, depth_vert = get_uv_depth(vertices, depth_np)
602
- #sfm_depth_np = get_SfM_depth(points3D, depth_np, gest_seg_np, K, R, t, 10)
603
- #uv, depth_vert = get_smooth_uv_depth(vertices, depth_np, gest_seg_np, sfm_depth_np)
604
- uv, depth_vert = get_smooth_uv_depth(vertices, depth_np, gest_seg_np, None)
605
 
606
  vertices_3d = uv_to_v3d(uv, depth_vert, K, R, t)
607
 
 
599
  continue
600
 
601
  #uv, depth_vert = get_uv_depth(vertices, depth_np)
602
+ sfm_depth_np = get_SfM_depth(points3D, depth_np, gest_seg_np, K, R, t, 0)
603
+ uv, depth_vert = get_smooth_uv_depth(vertices, depth_np, gest_seg_np, sfm_depth_np)
604
+ #uv, depth_vert = get_smooth_uv_depth(vertices, depth_np, gest_seg_np, None)
605
 
606
  vertices_3d = uv_to_v3d(uv, depth_vert, K, R, t)
607