HongChuang/ultralytics-ascend
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 3

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
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):