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:
parent
8136f66bfc
commit
69cfc8aa22
5 changed files with 11 additions and 10 deletions
|
|
@ -1,6 +1,6 @@
|
|||
# Ultralytics YOLO 🚀, AGPL-3.0 license
|
||||
|
||||
__version__ = "8.2.44"
|
||||
__version__ = "8.2.45"
|
||||
|
||||
import os
|
||||
|
||||
|
|
|
|||
|
|
@ -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."
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue