HongChuang/ultralytics-ascend
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 3

Update OBB predict examples with boats.jpg (#17052)

Browse source 
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
Co-authored-by: Jan Knobloch <116908874+jk4e@users.noreply.github.com>
Co-authored-by: Ultralytics Assistant <135830346+UltralyticsAssistant@users.noreply.github.com>
This commit is contained in:
Muhammad Rizwan Munawar 2024-10-26 00:19:57 +05:00 • committed by GitHub
parent b5fd7f3378
commit 1d6fd4c3cf
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
4 changed files with 8 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

2
docs/en/modes/predict.md
View file

@ -665,7 +665,7 @@ For more details see the [`Probs` class documentation](../reference/engine/resul
model = YOLO("yolo11n-obb.pt") model = YOLO("yolo11n-obb.pt")
# Run inference on an image # Run inference on an image
results = model("bus.jpg") # results list results = model("boats.jpg") # results list
# View results # View results
for r in results: for r in results:

6
docs/en/tasks/obb.md
View file

@ -141,14 +141,14 @@ Use a trained YOLO11n-obb model to run predictions on images.
model = YOLO("path/to/best.pt") # load a custom model model = YOLO("path/to/best.pt") # load a custom model
# Predict with the model # Predict with the model
results = model("https://ultralytics.com/images/bus.jpg") # predict on an image results = model("https://ultralytics.com/images/boats.jpg") # predict on an image
``` ```
=== "CLI" === "CLI"
```bash ```bash
yolo obb predict model=yolo11n-obb.pt source='https://ultralytics.com/images/bus.jpg' # predict with official model yolo obb predict model=yolo11n-obb.pt source='https://ultralytics.com/images/boats.jpg' # predict with official model
yolo obb predict model=path/to/best.pt source='https://ultralytics.com/images/bus.jpg' # predict with custom model yolo obb predict model=path/to/best.pt source='https://ultralytics.com/images/boats.jpg' # predict with custom model
``` ```
<p align="center"> <p align="center">

2
examples/tutorial.ipynb
View file

@ -583,7 +583,7 @@
"\n", "\n",
"model = YOLO('yolo11n-obb.pt') # load a pretrained YOLO OBB model\n", "model = YOLO('yolo11n-obb.pt') # load a pretrained YOLO OBB model\n",
"model.train(data='dota8.yaml', epochs=3) # train the model\n", "model.train(data='dota8.yaml', epochs=3) # train the model\n",
"model('https://ultralytics.com/images/bus.jpg') # predict on an image" "model('https://ultralytics.com/images/boats.jpg') # predict on an image"
], ],
"metadata": { "metadata": {
"id": "IJNKClOOB5YS" "id": "IJNKClOOB5YS"

4
ultralytics/cfg/__init__.py
View file

@ -809,7 +809,9 @@ def entrypoint(debug=""):
# Mode # Mode
if mode in {"predict", "track"} and "source" not in overrides: if mode in {"predict", "track"} and "source" not in overrides:
overrides["source"] = DEFAULT_CFG.source or ASSETS overrides["source"] = (
"https://ultralytics.com/images/boats.jpg" if task == "obb" else DEFAULT_CFG.source or ASSETS
)
LOGGER.warning(f"WARNING ⚠️ 'source' argument is missing. Using default 'source={overrides['source']}'.") LOGGER.warning(f"WARNING ⚠️ 'source' argument is missing. Using default 'source={overrides['source']}'.")
elif mode in {"train", "val"}: elif mode in {"train", "val"}:
if "data" not in overrides and "resume" not in overrides: if "data" not in overrides and "resume" not in overrides: