From d8564a48e124a4df872cbdb2023f42acd379b973 Mon Sep 17 00:00:00 2001 From: UltralyticsAssistant Date: Thu, 12 Sep 2024 09:00:03 +0000 Subject: [PATCH] Auto-format by https://ultralytics.com/actions --- ultralytics/engine/exporter.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/ultralytics/engine/exporter.py b/ultralytics/engine/exporter.py index f35043aee..123e3dd17 100644 --- a/ultralytics/engine/exporter.py +++ b/ultralytics/engine/exporter.py @@ -1309,9 +1309,9 @@ class Exporter: model = ct.models.MLModel(pipeline.spec, weights_dir=weights_dir) 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)" LOGGER.info(f"{prefix} pipeline success")