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

Reformat Markdown code blocks (#12795)

Browse source 
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
This commit is contained in:
Glenn Jocher 2024-05-18 18:58:06 +02:00 • committed by GitHub
parent 2af71d15a6
commit fceea033ad
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
128 changed files with 1067 additions and 1018 deletions
Download patch file Download diff file Expand all files Collapse all files

12
docs/en/integrations/tensorrt.md
View file

@ -85,16 +85,16 @@ Before diving into the usage instructions, be sure to check out the range of [YO
from ultralytics import YOLO
# Load the YOLOv8 model
model = YOLO('yolov8n.pt')
model = YOLO("yolov8n.pt")
# Export the model to TensorRT format
model.export(format='engine') # creates 'yolov8n.engine'
model.export(format="engine") # creates 'yolov8n.engine'
# Load the exported TensorRT model
tensorrt_model = YOLO('yolov8n.engine')
tensorrt_model = YOLO("yolov8n.engine")
# Run inference
results = tensorrt_model('https://ultralytics.com/images/bus.jpg')
results = tensorrt_model("https://ultralytics.com/images/bus.jpg")
```
=== "CLI"
@ -434,7 +434,7 @@ Expand sections below for information on how these models were exported and test
result = model.predict(
[img] * 8, # batch=8 of the same image
verbose=False,
device="cuda"
device="cuda",
)
```
@ -451,7 +451,7 @@ Expand sections below for information on how these models were exported and test
batch=1,
imgsz=640,
verbose=False,
device="cuda"
device="cuda",
)
```