pull/16/head
triple-Mu 2 years ago
parent 1478775411
commit dbeccf52a9
  1. 4
      docs/Normal.md

@ -12,7 +12,7 @@ from ultralytics import YOLO
# Load a model # Load a model
model = YOLO("yolov8n.pt") # load a pretrained model (recommended for training) model = YOLO("yolov8n.pt") # load a pretrained model (recommended for training)
success = model.export(format="engine") # export the model to engine format success = model.export(format="engine", device=0) # export the model to engine format
assert success assert success
``` ```
@ -21,7 +21,7 @@ After executing the above script, you will get an engine named `yolov8n.engine`
### 2. CLI tools ### 2. CLI tools
```shell ```shell
yolo export model=yolov8n.pt format=engine yolo export model=yolov8n.pt format=engine device=0
``` ```
After executing the above command, you will get an engine named `yolov8n.engine` too. After executing the above command, you will get an engine named `yolov8n.engine` too.

Loading…
Cancel
Save