mirror of
https://github.com/THU-MIG/yolov10.git
synced 2025-05-24 22:54:21 +08:00
620f3eb218
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com> Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
18 lines
823 B
Python
18 lines
823 B
Python
import importlib
|
|
import sys
|
|
|
|
from ultralytics.utils import LOGGER
|
|
|
|
# Set modules in sys.modules under their old name
|
|
sys.modules['ultralytics.yolo.data'] = importlib.import_module('ultralytics.data')
|
|
# This is for updating old cls models, or the way in following warning won't work.
|
|
sys.modules['ultralytics.yolo.data.augment'] = importlib.import_module('ultralytics.data.augment')
|
|
|
|
DATA_WARNING = """WARNING ⚠️ 'ultralytics.yolo.data' is deprecated since '8.0.136' and will be removed in '8.1.0'. Please use 'ultralytics.data' instead.
|
|
Note this warning may be related to loading older models. You can update your model to current structure with:
|
|
import torch
|
|
ckpt = torch.load("model.pt") # applies to both official and custom models
|
|
torch.save(ckpt, "updated-model.pt")
|
|
"""
|
|
LOGGER.warning(DATA_WARNING)
|