xclele/yolov10
1
0
Fork 0
mirror of https://github.com/THU-MIG/yolov10.git synced 2025-05-24 05:55:51 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix FastSAM text prompt results (#6510)

Browse Source
This commit is contained in:
Sulthan Suresh Fazeela 2023-11-22 16:30:30 +01:00 committed by GitHub
parent 830037e0ce
commit 0c4e97443b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
ultralytics/models/fastsam/prompt.py
View File

@ -342,7 +342,7 @@ class FastSAMPrompt:
max_idx = scores.argsort() max_idx = scores.argsort()
max_idx = max_idx[-1] max_idx = max_idx[-1]
max_idx += sum(np.array(filter_id) <= int(max_idx)) max_idx += sum(np.array(filter_id) <= int(max_idx))
self.results[0].masks.data = torch.tensor(np.array([ann['segmentation'] for ann in annotations])) self.results[0].masks.data = torch.tensor(np.array([annotations[max_idx]['segmentation']]))
return self.results return self.results
def everything_prompt(self): def everything_prompt(self):