Remove experimental compression builds from the release repository
Browse files- README.md +0 -2
- experiments/manifest.json +0 -30
- experiments/v2.5.0/detect/yolov8n-weight-compression/rk3588/fp16/metadata.yaml +0 -89
- experiments/v2.5.0/detect/yolov8n-weight-compression/rk3588/fp16/model.rknn +0 -3
- experiments/v2.5.0/detect/yolov8n-weight-compression/rk3588/int8/metadata.yaml +0 -89
- experiments/v2.5.0/detect/yolov8n-weight-compression/rk3588/int8/model.rknn +0 -3
- source/compress-weights.py +0 -48
README.md
CHANGED
|
@@ -27,5 +27,3 @@ compiled with the Model Zoo converter. The exact checkpoint URLs, exporter
|
|
| 27 |
commits, scripts, and ONNX graphs are under `source/`.
|
| 28 |
|
| 29 |
Models are organized by RKNN Toolkit2 version under `models/v<version>/`. Each model and its metadata share a folder.
|
| 30 |
-
|
| 31 |
-
The RK3588 weight compression trials are archived under [experiments](tree/main/experiments).
|
|
|
|
| 27 |
commits, scripts, and ONNX graphs are under `source/`.
|
| 28 |
|
| 29 |
Models are organized by RKNN Toolkit2 version under `models/v<version>/`. Each model and its metadata share a folder.
|
|
|
|
|
|
experiments/manifest.json
DELETED
|
@@ -1,30 +0,0 @@
|
|
| 1 |
-
[
|
| 2 |
-
{
|
| 3 |
-
"name": "yolov8n-weight-compression",
|
| 4 |
-
"family": "optimization",
|
| 5 |
-
"task": "detect",
|
| 6 |
-
"chip": "rk3588",
|
| 7 |
-
"precision": "fp16",
|
| 8 |
-
"path": "experiments/v2.5.0/detect/yolov8n-weight-compression/rk3588/fp16",
|
| 9 |
-
"sha256": "6375c7275e2aa16ddc1639fca085b3f6d323f71180af07b82c2fc070e00eb410",
|
| 10 |
-
"toolkit_version": "2.5.0",
|
| 11 |
-
"source_url": "https://github.com/airockchip/rknn_model_zoo/tree/main/examples/yolov8",
|
| 12 |
-
"upstream_project": "https://github.com/airockchip/rknn_model_zoo/tree/main/examples/yolov8",
|
| 13 |
-
"conversion": "experimental-compression",
|
| 14 |
-
"metadata_sha256": "48afa4c902e0f020d5ffafdadcf3b7e799e9986cdb16e2787114908da2f4e5b5"
|
| 15 |
-
},
|
| 16 |
-
{
|
| 17 |
-
"name": "yolov8n-weight-compression",
|
| 18 |
-
"family": "optimization",
|
| 19 |
-
"task": "detect",
|
| 20 |
-
"chip": "rk3588",
|
| 21 |
-
"precision": "int8",
|
| 22 |
-
"path": "experiments/v2.5.0/detect/yolov8n-weight-compression/rk3588/int8",
|
| 23 |
-
"sha256": "f3806db3444fd9034f6ed7f418e3b47b3a57df8a681e8df3859ced98fcb56f0d",
|
| 24 |
-
"toolkit_version": "2.5.0",
|
| 25 |
-
"source_url": "https://github.com/airockchip/rknn_model_zoo/tree/main/examples/yolov8",
|
| 26 |
-
"upstream_project": "https://github.com/airockchip/rknn_model_zoo/tree/main/examples/yolov8",
|
| 27 |
-
"conversion": "experimental-compression",
|
| 28 |
-
"metadata_sha256": "48afa4c902e0f020d5ffafdadcf3b7e799e9986cdb16e2787114908da2f4e5b5"
|
| 29 |
-
}
|
| 30 |
-
]
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
experiments/v2.5.0/detect/yolov8n-weight-compression/rk3588/fp16/metadata.yaml
DELETED
|
@@ -1,89 +0,0 @@
|
|
| 1 |
-
task: detect
|
| 2 |
-
batch: 1
|
| 3 |
-
channels: 3
|
| 4 |
-
imgsz:
|
| 5 |
-
- 640
|
| 6 |
-
- 640
|
| 7 |
-
names:
|
| 8 |
-
- person
|
| 9 |
-
- bicycle
|
| 10 |
-
- car
|
| 11 |
-
- motorcycle
|
| 12 |
-
- airplane
|
| 13 |
-
- bus
|
| 14 |
-
- train
|
| 15 |
-
- truck
|
| 16 |
-
- boat
|
| 17 |
-
- traffic light
|
| 18 |
-
- fire hydrant
|
| 19 |
-
- stop sign
|
| 20 |
-
- parking meter
|
| 21 |
-
- bench
|
| 22 |
-
- bird
|
| 23 |
-
- cat
|
| 24 |
-
- dog
|
| 25 |
-
- horse
|
| 26 |
-
- sheep
|
| 27 |
-
- cow
|
| 28 |
-
- elephant
|
| 29 |
-
- bear
|
| 30 |
-
- zebra
|
| 31 |
-
- giraffe
|
| 32 |
-
- backpack
|
| 33 |
-
- umbrella
|
| 34 |
-
- handbag
|
| 35 |
-
- tie
|
| 36 |
-
- suitcase
|
| 37 |
-
- frisbee
|
| 38 |
-
- skis
|
| 39 |
-
- snowboard
|
| 40 |
-
- sports ball
|
| 41 |
-
- kite
|
| 42 |
-
- baseball bat
|
| 43 |
-
- baseball glove
|
| 44 |
-
- skateboard
|
| 45 |
-
- surfboard
|
| 46 |
-
- tennis racket
|
| 47 |
-
- bottle
|
| 48 |
-
- wine glass
|
| 49 |
-
- cup
|
| 50 |
-
- fork
|
| 51 |
-
- knife
|
| 52 |
-
- spoon
|
| 53 |
-
- bowl
|
| 54 |
-
- banana
|
| 55 |
-
- apple
|
| 56 |
-
- sandwich
|
| 57 |
-
- orange
|
| 58 |
-
- broccoli
|
| 59 |
-
- carrot
|
| 60 |
-
- hot dog
|
| 61 |
-
- pizza
|
| 62 |
-
- donut
|
| 63 |
-
- cake
|
| 64 |
-
- chair
|
| 65 |
-
- couch
|
| 66 |
-
- potted plant
|
| 67 |
-
- bed
|
| 68 |
-
- dining table
|
| 69 |
-
- toilet
|
| 70 |
-
- tv
|
| 71 |
-
- laptop
|
| 72 |
-
- mouse
|
| 73 |
-
- remote
|
| 74 |
-
- keyboard
|
| 75 |
-
- cell phone
|
| 76 |
-
- microwave
|
| 77 |
-
- oven
|
| 78 |
-
- toaster
|
| 79 |
-
- sink
|
| 80 |
-
- refrigerator
|
| 81 |
-
- book
|
| 82 |
-
- clock
|
| 83 |
-
- vase
|
| 84 |
-
- scissors
|
| 85 |
-
- teddy bear
|
| 86 |
-
- hair drier
|
| 87 |
-
- toothbrush
|
| 88 |
-
target_platform: rk3588
|
| 89 |
-
coordinates: pixels
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
experiments/v2.5.0/detect/yolov8n-weight-compression/rk3588/fp16/model.rknn
DELETED
|
@@ -1,3 +0,0 @@
|
|
| 1 |
-
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:6375c7275e2aa16ddc1639fca085b3f6d323f71180af07b82c2fc070e00eb410
|
| 3 |
-
size 7666569
|
|
|
|
|
|
|
|
|
|
|
|
experiments/v2.5.0/detect/yolov8n-weight-compression/rk3588/int8/metadata.yaml
DELETED
|
@@ -1,89 +0,0 @@
|
|
| 1 |
-
task: detect
|
| 2 |
-
batch: 1
|
| 3 |
-
channels: 3
|
| 4 |
-
imgsz:
|
| 5 |
-
- 640
|
| 6 |
-
- 640
|
| 7 |
-
names:
|
| 8 |
-
- person
|
| 9 |
-
- bicycle
|
| 10 |
-
- car
|
| 11 |
-
- motorcycle
|
| 12 |
-
- airplane
|
| 13 |
-
- bus
|
| 14 |
-
- train
|
| 15 |
-
- truck
|
| 16 |
-
- boat
|
| 17 |
-
- traffic light
|
| 18 |
-
- fire hydrant
|
| 19 |
-
- stop sign
|
| 20 |
-
- parking meter
|
| 21 |
-
- bench
|
| 22 |
-
- bird
|
| 23 |
-
- cat
|
| 24 |
-
- dog
|
| 25 |
-
- horse
|
| 26 |
-
- sheep
|
| 27 |
-
- cow
|
| 28 |
-
- elephant
|
| 29 |
-
- bear
|
| 30 |
-
- zebra
|
| 31 |
-
- giraffe
|
| 32 |
-
- backpack
|
| 33 |
-
- umbrella
|
| 34 |
-
- handbag
|
| 35 |
-
- tie
|
| 36 |
-
- suitcase
|
| 37 |
-
- frisbee
|
| 38 |
-
- skis
|
| 39 |
-
- snowboard
|
| 40 |
-
- sports ball
|
| 41 |
-
- kite
|
| 42 |
-
- baseball bat
|
| 43 |
-
- baseball glove
|
| 44 |
-
- skateboard
|
| 45 |
-
- surfboard
|
| 46 |
-
- tennis racket
|
| 47 |
-
- bottle
|
| 48 |
-
- wine glass
|
| 49 |
-
- cup
|
| 50 |
-
- fork
|
| 51 |
-
- knife
|
| 52 |
-
- spoon
|
| 53 |
-
- bowl
|
| 54 |
-
- banana
|
| 55 |
-
- apple
|
| 56 |
-
- sandwich
|
| 57 |
-
- orange
|
| 58 |
-
- broccoli
|
| 59 |
-
- carrot
|
| 60 |
-
- hot dog
|
| 61 |
-
- pizza
|
| 62 |
-
- donut
|
| 63 |
-
- cake
|
| 64 |
-
- chair
|
| 65 |
-
- couch
|
| 66 |
-
- potted plant
|
| 67 |
-
- bed
|
| 68 |
-
- dining table
|
| 69 |
-
- toilet
|
| 70 |
-
- tv
|
| 71 |
-
- laptop
|
| 72 |
-
- mouse
|
| 73 |
-
- remote
|
| 74 |
-
- keyboard
|
| 75 |
-
- cell phone
|
| 76 |
-
- microwave
|
| 77 |
-
- oven
|
| 78 |
-
- toaster
|
| 79 |
-
- sink
|
| 80 |
-
- refrigerator
|
| 81 |
-
- book
|
| 82 |
-
- clock
|
| 83 |
-
- vase
|
| 84 |
-
- scissors
|
| 85 |
-
- teddy bear
|
| 86 |
-
- hair drier
|
| 87 |
-
- toothbrush
|
| 88 |
-
target_platform: rk3588
|
| 89 |
-
coordinates: pixels
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
experiments/v2.5.0/detect/yolov8n-weight-compression/rk3588/int8/model.rknn
DELETED
|
@@ -1,3 +0,0 @@
|
|
| 1 |
-
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:f3806db3444fd9034f6ed7f418e3b47b3a57df8a681e8df3859ced98fcb56f0d
|
| 3 |
-
size 4317323
|
|
|
|
|
|
|
|
|
|
|
|
source/compress-weights.py
DELETED
|
@@ -1,48 +0,0 @@
|
|
| 1 |
-
"""Try RK3588 weight compression with the original converters and calibration."""
|
| 2 |
-
from pathlib import Path
|
| 3 |
-
import os
|
| 4 |
-
import runpy
|
| 5 |
-
import shutil
|
| 6 |
-
import sys
|
| 7 |
-
|
| 8 |
-
from rknn.api import RKNN
|
| 9 |
-
|
| 10 |
-
workspace = Path.home() / "hanzos_workspace"
|
| 11 |
-
matrix = workspace / "nvr-matrix-20261107"
|
| 12 |
-
zoo = workspace / "rknn_model_zoo"
|
| 13 |
-
exports = workspace / "yolo-rknn"
|
| 14 |
-
source, precision = sys.argv[1:]
|
| 15 |
-
output = exports / "optimization/rk3588/yolov8n" / source / precision
|
| 16 |
-
output.parent.mkdir(parents=True, exist_ok=True)
|
| 17 |
-
config = RKNN.config
|
| 18 |
-
|
| 19 |
-
|
| 20 |
-
def compress(self, *args, **kwargs):
|
| 21 |
-
return config(self, *args, **dict(kwargs, compress_weight=True))
|
| 22 |
-
|
| 23 |
-
|
| 24 |
-
RKNN.config = compress
|
| 25 |
-
|
| 26 |
-
if source == "rockchip":
|
| 27 |
-
original_precision = {"fp16": "fp", "int8": "i8"}[precision]
|
| 28 |
-
original = exports / "rknn/rk3588/yolov8/yolov8n" / original_precision
|
| 29 |
-
example = zoo / "examples/yolov8/python"
|
| 30 |
-
output.mkdir()
|
| 31 |
-
os.chdir(example)
|
| 32 |
-
converter = example / "convert.py"
|
| 33 |
-
sys.argv = [str(converter), str(exports / "onnx/yolov8/yolov8n.onnx"), "rk3588",
|
| 34 |
-
original_precision, str(output / "model.rknn")]
|
| 35 |
-
runpy.run_path(str(converter), run_name="__main__")
|
| 36 |
-
shutil.copy2(original / "metadata.yaml", output)
|
| 37 |
-
elif source == "ultralytics":
|
| 38 |
-
converter = matrix / "tools/convert-model.py"
|
| 39 |
-
sys.path.insert(0, str(converter.parent))
|
| 40 |
-
sys.argv = [str(converter), str(matrix / "weights/yolov8n.pt"), "--chip", "rk3588",
|
| 41 |
-
"--precision", precision, "--output", str(output)]
|
| 42 |
-
if precision == "int8":
|
| 43 |
-
sys.argv += ["--data", str(matrix / "calibration/coco128.yaml")]
|
| 44 |
-
runpy.run_path(str(converter), run_name="__main__")
|
| 45 |
-
else:
|
| 46 |
-
raise ValueError(source)
|
| 47 |
-
|
| 48 |
-
assert (output / "model.rknn").stat().st_size > 0
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|