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

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

Browse Source
This commit is contained in:
Glenn Jocher
2022-12-29 14:58:03 +01:00
committed by GitHub
parent 92dad1c1b5
commit 37e0f6e0e4
8 changed files with 8 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
ultralytics/yolo/v8/segment/predict.py
View File

@ -1,8 +1,7 @@
import hydra
import torch
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.plotting import colors, save_one_box
from ..detect.predict import DetectionPredictor

2
ultralytics/yolo/v8/segment/train.py
View File

@ -5,7 +5,7 @@ import torch.nn.functional as F
from ultralytics.nn.tasks import SegmentationModel
from ultralytics.yolo import v8
from ultralytics.yolo.engine.trainer import DEFAULT_CONFIG
from ultralytics.yolo.utils import DEFAULT_CONFIG
from ultralytics.yolo.utils.loss import BboxLoss
from ultralytics.yolo.utils.ops import crop_mask, xywh2xyxy, xyxy2xywh
from ultralytics.yolo.utils.plotting import plot_images, plot_results

3
ultralytics/yolo/v8/segment/val.py
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