Skip to content

Commit

Permalink
Browse files Browse the repository at this point in the history
  • Loading branch information
UltralyticsAssistant committed Feb 29, 2024
1 parent 5399154 commit bd9c235
Show file tree
Hide file tree
Showing 6 changed files with 11 additions and 6 deletions.
1 change: 1 addition & 0 deletions docs/build_docs.py
Original file line number Diff line number Diff line change
Expand Up @@ -23,6 +23,7 @@
Note:
- This script is built to be run in an environment where Python and MkDocs are installed and properly configured.
"""

import os
import re
import shutil
Expand Down
6 changes: 3 additions & 3 deletions ultralytics/data/explorer/gui/dash.py
Original file line number Diff line number Diff line change
Expand Up @@ -144,9 +144,9 @@ def run_sql_query():
def run_ai_query():
"""Execute SQL query and update session state with query results."""
if not SETTINGS["openai_api_key"]:
st.session_state[
"error"
] = 'OpenAI API key not found in settings. Please run yolo settings openai_api_key="..."'
st.session_state["error"] = (
'OpenAI API key not found in settings. Please run yolo settings openai_api_key="..."'
)
return
st.session_state["error"] = None
query = st.session_state.get("ai_query")
Expand Down
7 changes: 4 additions & 3 deletions ultralytics/engine/exporter.py
Original file line number Diff line number Diff line change
Expand Up @@ -48,6 +48,7 @@
$ ln -s ../../yolov5/yolov8n_web_model public/yolov8n_web_model
$ npm start
"""

import json
import os
import shutil
Expand Down Expand Up @@ -1079,9 +1080,9 @@ def _pipeline_coreml(self, model, weights_dir=None, prefix=colorstr("CoreML Pipe
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")
Expand Down
1 change: 1 addition & 0 deletions ultralytics/engine/predictor.py
Original file line number Diff line number Diff line change
Expand Up @@ -27,6 +27,7 @@
yolov8n_edgetpu.tflite # TensorFlow Edge TPU
yolov8n_paddle_model # PaddlePaddle
"""

import platform
import threading
from pathlib import Path
Expand Down
1 change: 1 addition & 0 deletions ultralytics/engine/tuner.py
Original file line number Diff line number Diff line change
Expand Up @@ -16,6 +16,7 @@
model.tune(data='coco8.yaml', epochs=10, iterations=300, optimizer='AdamW', plots=False, save=False, val=False)
```
"""

import random
import shutil
import subprocess
Expand Down
1 change: 1 addition & 0 deletions ultralytics/engine/validator.py
Original file line number Diff line number Diff line change
Expand Up @@ -18,6 +18,7 @@
yolov8n_edgetpu.tflite # TensorFlow Edge TPU
yolov8n_paddle_model # PaddlePaddle
"""

import json
import time
from pathlib import Path
Expand Down

0 comments on commit bd9c235

Please sign in to comment.