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

ultralytics 8.0.98 add Baidu RT-DETR models (#2527)

Browse source 
Co-authored-by: Kalen Michael <kalenmike@gmail.com>
Co-authored-by: Laughing <61612323+Laughing-q@users.noreply.github.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Yonghye Kwon <developer.0hye@gmail.com>
Co-authored-by: Dowon <ks2515@naver.com>
This commit is contained in:
Glenn Jocher 2023-05-11 01:15:20 +02:00 • committed by GitHub
parent d1107ca4cb
commit 229119c376
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
33 changed files with 2172 additions and 834 deletions
Download patch file Download diff file Expand all files Collapse all files

2
ultralytics/vit/sam/modules/decoders.py
View file

@ -157,5 +157,5 @@ class MLP(nn.Module):
for i, layer in enumerate(self.layers):
x = F.relu(layer(x)) if i < self.num_layers - 1 else layer(x)
if self.sigmoid_output:
x = F.sigmoid(x)
x = torch.sigmoid(x)
return x