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

[RTDETR]Fix val loss (#3280)

Browse source
This commit is contained in:
Laughing 2023-06-20 23:59:04 +08:00 • committed by GitHub
parent d8701b42ca
commit 9d1e5567de
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
4 changed files with 12 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

2
ultralytics/nn/tasks.py
View file

@ -432,7 +432,7 @@ class RTDETRDetectionModel(DetectionModel):
'gt_groups': gt_groups}
preds = self.predict(img, batch=targets) if preds is None else preds
dec_bboxes, dec_scores, enc_bboxes, enc_scores, dn_meta = preds
dec_bboxes, dec_scores, enc_bboxes, enc_scores, dn_meta = preds if self.training else preds[1]
if dn_meta is None:
dn_bboxes, dn_scores = None, None
else: