Improve model card and surface related exports
Browse files
README.md
CHANGED
|
@@ -10,24 +10,34 @@ tags:
|
|
| 10 |
|
| 11 |
# YOLOv8-RKNN2-Ultralytics
|
| 12 |
|
| 13 |
-
|
| 14 |
|
| 15 |
-
|
| 16 |
|
| 17 |
-
##
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 18 |
|
| 19 |
-
|
| 20 |
|
| 21 |
-
|
| 22 |
|
| 23 |
-
|
| 24 |
|
| 25 |
-
##
|
| 26 |
|
| 27 |
-
|
| 28 |
|
| 29 |
-
|
|
|
|
|
|
|
| 30 |
|
| 31 |
-
|
|
|
|
|
|
|
| 32 |
|
| 33 |
-
|
|
|
|
| 10 |
|
| 11 |
# YOLOv8-RKNN2-Ultralytics
|
| 12 |
|
| 13 |
+
> **Looking for the Rockchip Model Zoo and custom exports?** See [YOLOv8-RKNN2](https://huggingface.co/HanzoHuang/YOLOv8-RKNN2).
|
| 14 |
|
| 15 |
+
100 FP16 and INT8 YOLOv8 exports for **RK3576** and **RK3588**, produced with the Ultralytics export workflow and task-specific RKNN conversion tools. Supported tasks include classification, detection, oriented bounding boxes, pose, and instance segmentation.
|
| 16 |
|
| 17 |
+
## Choose a model
|
| 18 |
+
|
| 19 |
+
Models are organized by toolkit version, task, variant, target chip, and precision:
|
| 20 |
+
|
| 21 |
+
```text
|
| 22 |
+
models/v<toolkit-version>/<task>/<variant>/<chip>/<precision>/
|
| 23 |
+
```
|
| 24 |
|
| 25 |
+
Download `model.rknn` and `metadata.yaml` from the same directory. The [manifest](manifest.json) lists the available models and checksums.
|
| 26 |
|
| 27 |
+
## INT8 detection
|
| 28 |
|
| 29 |
+
Use the `metadata.yaml` from the same model directory. INT8 detection outputs normalized box coordinates and requires `coordinates: normalized`. Import both files together in the warehouse, or keep them in the same directory for the CLI. Treating normalized coordinates as pixels can result in no boxes.
|
| 30 |
|
| 31 |
+
## Reproduce the conversion
|
| 32 |
|
| 33 |
+
The [source directory](tree/main/source) contains the original checkpoints, conversion scripts, and environment and calibration records. Conversion used [RKNN Toolkit2 2.3.2](https://github.com/airockchip/rknn-toolkit2); upstream Ultralytics source is available on [PyPI](https://pypi.org/project/ultralytics/#history).
|
| 34 |
|
| 35 |
+
```bash
|
| 36 |
+
python source/convert-model.py --help
|
| 37 |
+
```
|
| 38 |
|
| 39 |
+
For INT8 detection, pass a COCO128 dataset YAML. For other INT8 tasks, pass a COCO128 image list. Use `--calibration-limit 128` and the input size listed in the manifest.
|
| 40 |
+
|
| 41 |
+
## Source and license
|
| 42 |
|
| 43 |
+
The inputs are official Ultralytics checkpoints. Conversion changes the format and quantizes the INT8 graphs. The RKNN files and original weights are distributed under AGPL-3.0; see [LICENSE](LICENSE) and the [checkpoint URLs and hashes](source/weights.json).
|