Import DEFAULT_CONFIG from utils/__init__.py (#115)

This commit is contained in:
Glenn Jocher 2022-12-29 14:58:03 +01:00 committed by GitHub
parent 92dad1c1b5
commit 37e0f6e0e4
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
8 changed files with 8 additions and 13 deletions

View file

@ -5,8 +5,7 @@ import numpy as np
import torch
import torch.nn.functional as F
from ultralytics.yolo.engine.trainer import DEFAULT_CONFIG
from ultralytics.yolo.utils import ops
from ultralytics.yolo.utils import DEFAULT_CONFIG, ops
from ultralytics.yolo.utils.metrics import ConfusionMatrix, SegmentMetrics, box_iou, mask_iou
from ultralytics.yolo.utils.plotting import output_to_target, plot_images