mirror of
https://github.com/ultralytics/ultralytics.git
synced 2025-09-15 15:48:41 +08:00
Merge 0cc8e07488 into 1aa3688613
This commit is contained in:
commit
e672e64bd0
@ -205,6 +205,9 @@ makei05@outlook.de:
|
||||
matthewnoyce@icloud.com:
|
||||
avatar: https://avatars.githubusercontent.com/u/131261051?v=4
|
||||
username: MatthewNoyce
|
||||
miles@ultralytics.com:
|
||||
avatar: null
|
||||
username: null
|
||||
muhammadravi251001@gmail.com:
|
||||
avatar: https://avatars.githubusercontent.com/u/78993021?v=4
|
||||
username: muhammadravi251001
|
||||
|
||||
@ -1037,7 +1037,6 @@ class YOLOEModel(DetectionModel):
|
||||
if without_reprta:
|
||||
return txt_feats
|
||||
|
||||
assert not self.training
|
||||
head = self.model[-1]
|
||||
assert isinstance(head, YOLOEDetect)
|
||||
return head.get_tpe(txt_feats) # run auxiliary text head
|
||||
|
||||
Loading…
Reference in New Issue
Block a user