Spaces:

NN-BRD
/

hackathon_depth_segment

Runtime error

jens commited on Aug 14, 2023

Commit

aa4b380

1 Parent(s): 1758fb9

test lbls SAM

Files changed (1) hide show

inference.py CHANGED Viewed

@@ -135,6 +135,7 @@ class SegmentPredictor:
         self.conditioned_pred.set_image(image)
     def cond_pred(self, pts, lbls):
         masks, _, _ = self.conditioned_pred.predict(
             point_coords=pts,
             point_labels=lbls,

         self.conditioned_pred.set_image(image)
     def cond_pred(self, pts, lbls):
+        print(lbls)
         masks, _, _ = self.conditioned_pred.predict(
             point_coords=pts,
             point_labels=lbls,