|
|
|
@ -12,7 +12,7 @@ from ultralytics import YOLO |
|
|
|
|
|
|
|
|
|
# Load a model |
|
|
|
|
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 |
|
|
|
|
``` |
|
|
|
|
|
|
|
|
@ -21,7 +21,7 @@ After executing the above script, you will get an engine named `yolov8n.engine` |
|
|
|
|
### 2. CLI tools |
|
|
|
|
|
|
|
|
|
```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. |
|
|
|
|