diff --git a/export.py b/export.py index 9ea2b93..8fe7ce8 100644 --- a/export.py +++ b/export.py @@ -742,9 +742,9 @@ def pipeline_coreml(model, im, file, names, y, prefix=colorstr("CoreML Pipeline: model = ct.models.MLModel(pipeline.spec) model.input_description["image"] = "Input image" model.input_description["iouThreshold"] = f"(optional) IOU Threshold override (default: {nms.iouThreshold})" - model.input_description[ - "confidenceThreshold" - ] = f"(optional) Confidence Threshold override (default: {nms.confidenceThreshold})" + model.input_description["confidenceThreshold"] = ( + f"(optional) Confidence Threshold override (default: {nms.confidenceThreshold})" + ) model.output_description["confidence"] = 'Boxes × Class confidence (see user-defined metadata "classes")' model.output_description["coordinates"] = "Boxes × [x, y, width, height] (relative to image size)" model.save(f) # pipelined diff --git a/models/experimental.py b/models/experimental.py index 834c720..6152cef 100644 --- a/models/experimental.py +++ b/models/experimental.py @@ -1,5 +1,6 @@ # YOLOv5 🚀 by Ultralytics, AGPL-3.0 license """Experimental modules.""" + import math import numpy as np diff --git a/utils/loggers/__init__.py b/utils/loggers/__init__.py index c3fbded..2a2c5d7 100644 --- a/utils/loggers/__init__.py +++ b/utils/loggers/__init__.py @@ -1,5 +1,6 @@ # YOLOv5 🚀 by Ultralytics, AGPL-3.0 license """Logging utils.""" + import json import os import warnings diff --git a/utils/loggers/clearml/clearml_utils.py b/utils/loggers/clearml/clearml_utils.py index 1bbea61..6a6ed76 100644 --- a/utils/loggers/clearml/clearml_utils.py +++ b/utils/loggers/clearml/clearml_utils.py @@ -1,4 +1,5 @@ """Main Logger class for ClearML experiment tracking.""" + import glob import re from pathlib import Path diff --git a/utils/loss.py b/utils/loss.py index e8563e1..9d09f9d 100644 --- a/utils/loss.py +++ b/utils/loss.py @@ -9,7 +9,7 @@ from utils.torch_utils import de_parallel def smooth_BCE(eps=0.1): - """Returns label smoothing BCE targets for reducing overfitting; pos: `1.0 - 0.5*eps`, neg: `0.5*eps`. For details see ttps://github.com/ultralytics/yolov3/issues/238#issuecomment-598028441""" + """Returns label smoothing BCE targets for reducing overfitting; pos: `1.0 - 0.5*eps`, neg: `0.5*eps`. For details see https://github.com/ultralytics/yolov3/issues/238#issuecomment-598028441""" return 1.0 - 0.5 * eps, 0.5 * eps