ultralytics 8.0.136
refactor and simplify package (#3748)
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>
This commit is contained in:
@ -5,7 +5,7 @@ from pathlib import Path
|
||||
|
||||
import pytest
|
||||
|
||||
from ultralytics.yolo.utils import ONLINE, ROOT, SETTINGS
|
||||
from ultralytics.utils import ONLINE, ROOT, SETTINGS
|
||||
|
||||
WEIGHT_DIR = Path(SETTINGS['weights_dir'])
|
||||
TASK_ARGS = [ # (task, model, data)
|
||||
|
@ -3,10 +3,10 @@
|
||||
from pathlib import Path
|
||||
|
||||
from ultralytics import YOLO
|
||||
from ultralytics.yolo.cfg import get_cfg
|
||||
from ultralytics.yolo.engine.exporter import Exporter
|
||||
from ultralytics.yolo.utils import DEFAULT_CFG, ROOT, SETTINGS
|
||||
from ultralytics.yolo.v8 import classify, detect, segment
|
||||
from ultralytics.cfg import get_cfg
|
||||
from ultralytics.engine.exporter import Exporter
|
||||
from ultralytics.models.yolo import classify, detect, segment
|
||||
from ultralytics.utils import DEFAULT_CFG, ROOT, SETTINGS
|
||||
|
||||
CFG_DET = 'yolov8n.yaml'
|
||||
CFG_SEG = 'yolov8n-seg.yaml'
|
||||
|
@ -9,8 +9,8 @@ from PIL import Image
|
||||
from torchvision.transforms import ToTensor
|
||||
|
||||
from ultralytics import RTDETR, YOLO
|
||||
from ultralytics.yolo.data.build import load_inference_source
|
||||
from ultralytics.yolo.utils import LINUX, ONLINE, ROOT, SETTINGS
|
||||
from ultralytics.data.build import load_inference_source
|
||||
from ultralytics.utils import LINUX, ONLINE, ROOT, SETTINGS
|
||||
|
||||
MODEL = Path(SETTINGS['weights_dir']) / 'yolov8n.pt'
|
||||
CFG = 'yolov8n.yaml'
|
||||
@ -102,7 +102,7 @@ def test_val_scratch():
|
||||
|
||||
def test_amp():
|
||||
if torch.cuda.is_available():
|
||||
from ultralytics.yolo.utils.checks import check_amp
|
||||
from ultralytics.utils.checks import check_amp
|
||||
model = YOLO(MODEL).model.cuda()
|
||||
assert check_amp(model)
|
||||
|
||||
|
Reference in New Issue
Block a user