Include scores in the G-SAM model response

landing-ai · Apr 4, 2024 · 8589f93 · 8589f93
1 parent b1f7bdc
commit 8589f93
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/vision_agent/tools/tools.py b/vision_agent/tools/tools.py
@@ -301,6 +301,7 @@ def __call__(self, prompt: str, image: Union[str, ImageType]) -> Dict:
  for mask in data["masks"]
  ]
  ret_pred["labels"] = data["labels"]
+ ret_pred["scores"] = data["scores"]
  return ret_pred