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

ultralytics 8.2.50 new Streamlit live inference Solution (#14210)

Browse source 
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: Muhammad Rizwan Munawar <muhammadrizwanmunawar123@gmail.com>
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
Co-authored-by: RizwanMunawar <chr043416@gmail.com>
Co-authored-by: Kayzwer <68285002+Kayzwer@users.noreply.github.com>
This commit is contained in:
Glenn Jocher 2024-07-05 22:02:38 +02:00 • committed by GitHub
parent 5f0fd710a4
commit 26a664f636
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
20 changed files with 350 additions and 22 deletions
Download patch file Download diff file Expand all files Collapse all files

2
ultralytics/models/sam/modules/tiny_encoder.py
View file

@ -261,7 +261,7 @@ class Attention(torch.nn.Module):
"""
super().__init__()
assert isinstance(resolution, tuple) and len(resolution) == 2
assert isinstance(resolution, tuple) and len(resolution) == 2, "'resolution' argument not tuple of length 2"
self.num_heads = num_heads
self.scale = key_dim**-0.5
self.key_dim = key_dim