xdobro23/YOLOv8-16bit
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

New ASSETS and trackers GMC cleanup (#4425)

Browse Source 
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
This commit is contained in:
Glenn Jocher
2023-08-17 18:19:05 +02:00
committed by GitHub
parent aaba14e6b2
commit 9d27e7ada4
32 changed files with 222 additions and 201 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
ultralytics/nn/tasks.py
View File

@ -303,13 +303,6 @@ class SegmentationModel(DetectionModel):
def init_criterion(self):
return v8SegmentationLoss(self)
def _predict_augment(self, x):
"""Perform augmentations on input image x and return augmented inference."""
LOGGER.warning(
f'WARNING ⚠️ {self.__class__.__name__} has not supported augment inference yet! Now using single-scale inference instead.'
)
return self._predict_once(x)
class PoseModel(DetectionModel):
"""YOLOv8 pose model."""
@ -326,13 +319,6 @@ class PoseModel(DetectionModel):
def init_criterion(self):
return v8PoseLoss(self)
def _predict_augment(self, x):
"""Perform augmentations on input image x and return augmented inference."""
LOGGER.warning(
f'WARNING ⚠️ {self.__class__.__name__} has not supported augment inference yet! Now using single-scale inference instead.'
)
return self._predict_once(x)
class ClassificationModel(BaseModel):
"""YOLOv8 classification model."""