ultralytics 8.2.45 Fix YOLOv8 augment inference (#14017)

Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
This commit is contained in:
Laughing 2024-06-27 00:28:43 +08:00 committed by GitHub
parent 8136f66bfc
commit 69cfc8aa22
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
5 changed files with 11 additions and 10 deletions

View file

@ -1,6 +1,6 @@
# Ultralytics YOLO 🚀, AGPL-3.0 license
__version__ = "8.2.44"
__version__ = "8.2.45"
import os

View file

@ -337,7 +337,7 @@ class DetectionModel(BaseModel):
def _predict_augment(self, x):
"""Perform augmentations on input image x and return augmented inference and train outputs."""
if self.end2end:
if getattr(self, "end2end", False):
LOGGER.warning(
"WARNING ⚠️ End2End model does not support 'augment=True' prediction. "
"Reverting to single-scale prediction."