diff --git a/export.py b/export.py index 8fe7ce8fdce1..9ea2b936d740 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 6152cef1b389..834c7201fbd4 100644 --- a/models/experimental.py +++ b/models/experimental.py @@ -1,6 +1,5 @@ # 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 2a2c5d734c2e..c3fbded50a3c 100644 --- a/utils/loggers/__init__.py +++ b/utils/loggers/__init__.py @@ -1,6 +1,5 @@ # 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 6a6ed7636c88..1bbea61effc2 100644 --- a/utils/loggers/clearml/clearml_utils.py +++ b/utils/loggers/clearml/clearml_utils.py @@ -1,5 +1,4 @@ """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 9d09f9df0261..e8563e186f40 100644 --- a/utils/loss.py +++ b/utils/loss.py @@ -9,7 +9,7 @@ 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 https://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 ttps://github.com/ultralytics/yolov3/issues/238#issuecomment-598028441""" return 1.0 - 0.5 * eps, 0.5 * eps