jounery-d commited on
Commit
4c44942
Β·
verified Β·
1 Parent(s): 4c08e56

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +100 -3
README.md CHANGED
@@ -1,3 +1,100 @@
1
- ---
2
- license: mit
3
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: mit
3
+ language:
4
+ - en
5
+ base_model:
6
+ - edsr
7
+ - espcn
8
+ pipeline_tag: speaker-recognition
9
+ tags:
10
+ - Image
11
+ - SuperResolution
12
+ ---
13
+
14
+ # SuperResolution
15
+
16
+ This version of SuperResolution has been converted to run on the Axera NPU using **w8a8** quantization.
17
+
18
+ This model has been optimized with the following LoRA:
19
+
20
+ Compatible with Pulsar2 version: 4.2
21
+
22
+ ## Convert tools links:
23
+
24
+ For those who are interested in model conversion, you can try to export axmodel through
25
+
26
+ - [The repo of AXera Platform](https://github.com/AXERA-TECH/ax-samples), which you can get the detail of guide
27
+
28
+ - [Pulsar2 Link, How to Convert ONNX to axmodel](https://pulsar2-docs.readthedocs.io/en/latest/pulsar2/introduction.html)
29
+
30
+
31
+ ## Support Platform
32
+
33
+ - AX650
34
+ - [M4N-Dock(爱芯派Pro)](https://wiki.sipeed.com/hardware/zh/maixIV/m4ndock/m4ndock.html)
35
+ - [M.2 Accelerator card](https://axcl-docs.readthedocs.io/zh-cn/latest/doc_guide_hardware.html)
36
+ - AX630C
37
+ - [爱芯派2](https://axera-pi-2-docs-cn.readthedocs.io/zh-cn/latest/index.html)
38
+ - [Module-LLM](https://docs.m5stack.com/zh_CN/module/Module-LLM)
39
+ - [LLM630 Compute Kit](https://docs.m5stack.com/zh_CN/core/LLM630%20Compute%20Kit)
40
+
41
+ |Chips|model|cost|
42
+ |--|--|--|
43
+ |AX650|EDSR|800 ms|
44
+ | |ESPCN|22 ms|
45
+
46
+ ## How to use
47
+
48
+ Download all files from this repository to the device
49
+
50
+ ```
51
+
52
+ root@ax650:~/SuperResolution# tree
53
+ .
54
+ |-- model_convert
55
+ | -- axmodel
56
+ | `-- edsr_baseline_x2_1.axmodel
57
+ | `-- espcn_x2_T9.axmodel
58
+ | -- onnx
59
+ | `-- edsr_baseline_x2_1.onnx
60
+ | `-- espcn_x2_T9.onnx
61
+ | `-- build_config_edsr.json
62
+ | `-- build_config_espcn.json
63
+ |-- python
64
+ | `-- run_onnx.py
65
+ | `-- run_axmodel.py
66
+ | `-- common.py
67
+ | `-- imgproc.py
68
+ |-- video
69
+ | `-- test_1920x1080.mp4
70
+ | `-- 1.png
71
+ | `-- 2.png
72
+
73
+
74
+ ```
75
+
76
+ ### Inference
77
+
78
+ Input Data:
79
+ |-- video
80
+ | `-- test_1920x1080.mp4
81
+
82
+ #### Inference with AX650 Host, such as M4N-Dock(爱芯派Pro)
83
+
84
+ ```
85
+ root@ax650 ~/SuperResolution #python3 run_axmodel.py --model ./edsr_baseline_x2_1.axmodel --scale 2 --dir_demo ./test_1920x1080.mp4
86
+ [INFO] Available providers: ['AxEngineExecutionProvider']
87
+ [INFO] Using provider: AxEngineExecutionProvider
88
+ [INFO] Chip type: ChipType.MC50
89
+ [INFO] VNPU type: VNPUType.DISABLED
90
+ [INFO] Engine version: 2.12.0s
91
+ [INFO] Model type: 2 (triple core)
92
+ [INFO] Compiler version: 4.2 6bff2f67
93
+ 100%|β–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆβ–ˆ| 267/267 [10:06<00:00, 2.27s/it]
94
+ Total time: 99.582 seconds for 267 frames
95
+ Average time: 0.373 seconds for each frame
96
+
97
+ ```
98
+
99
+ Output:
100
+ [INFO]: The