xclele/yolov10
1
0
Fork 0
mirror of https://github.com/THU-MIG/yolov10.git synced 2025-05-23 21:44:22 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

[RTDETR]fix mosaic<1.0 (#3518)

Browse Source
This commit is contained in:
Laughing 2023-07-04 16:09:17 +08:00 committed by GitHub
parent 31b46bf2b4
commit 8c0a7440df
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
ultralytics/yolo/data/augment.py
View File

@ -427,7 +427,7 @@ class RandomPerspective:
""" """
if self.pre_transform and 'mosaic_border' not in labels: if self.pre_transform and 'mosaic_border' not in labels:
labels = self.pre_transform(labels) labels = self.pre_transform(labels)
labels.pop('ratio_pad') # do not need ratio pad labels.pop('ratio_pad', None) # do not need ratio pad
img = labels['img'] img = labels['img']
cls = labels['cls'] cls = labels['cls']