Skip to content

Commit

Permalink
Update lang_sam.py
Browse files Browse the repository at this point in the history
  • Loading branch information
luca-medeiros committed Apr 25, 2023
1 parent 036eabd commit 542fbb2
Showing 1 changed file with 0 additions and 3 deletions.
3 changes: 0 additions & 3 deletions lang_sam/lang_sam.py
Original file line number Diff line number Diff line change
Expand Up @@ -112,6 +112,3 @@ def predict(self, image_pil, text_prompt, box_threshold=0.3, text_threshold=0.25
masks = self.predict_sam(image_pil, boxes)
masks = masks.squeeze(1)
return masks, boxes, phrases, logits


LangSAM()

0 comments on commit 542fbb2

Please sign in to comment.