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

Add Export Arguments tables to all Export docs (#18952)

Browse source 
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
Co-authored-by: Francesco Mattioli <Francesco.mttl@gmail.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
This commit is contained in:
Lakshantha Dissanayake 2025-02-05 18:43:02 -08:00 • committed by GitHub
parent c0a4201370
commit 39b037408f
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
16 changed files with 162 additions and 33 deletions
Download patch file Download diff file Expand all files Collapse all files

10
docs/en/integrations/torchscript.md
View file

@ -105,6 +105,16 @@ Before diving into the usage instructions, it's important to note that while all
yolo predict model=yolo11n.torchscript source='https://ultralytics.com/images/bus.jpg'
```
### Export Arguments
| Argument | Type | Default | Description |
| ---------- | ---------------- | ------------- | --------------------------------------------------------------------------------------------------------------------------------------- |
| `format` | `str` | `torchscript` | Target format for the exported model, defining compatibility with various deployment environments. |
| `imgsz` | `int` or `tuple` | `640` | Desired image size for the model input. Can be an integer for square images or a tuple `(height, width)` for specific dimensions. |
| `optimize` | `bool` | `False` | Applies optimization for mobile devices, potentially reducing model size and improving performance. |
| `nms` | `bool` | `False` | Adds Non-Maximum Suppression (NMS), essential for accurate and efficient detection post-processing. |
| `batch` | `int` | `1` | Specifies export model batch inference size or the max number of images the exported model will process concurrently in `predict` mode. |
For more details about the export process, visit the [Ultralytics documentation page on exporting](../modes/export.md).
## Deploying Exported YOLO11 TorchScript Models