HanzoHuang commited on
Commit
b49c154
·
verified ·
1 Parent(s): 80413f6

Remove experimental compression builds from the release repository

Browse files
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