improve accuracy

deng232 · Aug 23, 2023 · afbe25d · afbe25d
1 parent 8f9f6b5
commit afbe25d
Show file tree

Hide file tree

Showing 2 changed files with 108 additions and 109 deletions.
diff --git a/DINO_SAM_Zoe.ipynb b/DINO_SAM_Zoe.ipynb
diff --git a/segdepth.py b/segdepth.py
@@ -37,9 +37,8 @@
 
 depth_map = model_zoe_nk.infer_pil(image, output_type="numpy")
 
-# list of pixles of a masked region in depth_map
-masklist = [depth_map[mask] for mask in detections[0].mask]
-
+#depth_map = model_zoe_nk.infer_pil(image, output_type="numpy")
+depth_map = model_zoe_nk.infer_pil(image, output_type="numpy",pad_input=False) # "pad_input = False" increase accuracy
 
 box_annotator = sv.BoxAnnotator(text_scale=1.5)
 mask_annotator = sv.MaskAnnotator()