wzf19947 commited on
Commit
42d4762
·
1 Parent(s): 4dca198

first commit

Browse files
Files changed (2) hide show
  1. MixDehazeNet/README.md +0 -89
  2. README.md +87 -1
MixDehazeNet/README.md DELETED
@@ -1,89 +0,0 @@
1
- ---
2
- license: mit
3
- language:
4
- - en
5
- pipeline_tag: image-to-image
6
- tags:
7
- - dehazing
8
- - ImageEnhancement
9
-
10
- # ImageDehazing
11
-
12
- This is a collection of image dehazing algorithms, models have been converted to run on the Axera NPU using **w8a8** quantization.
13
-
14
- This model has been optimized with the following LoRA:
15
-
16
- Compatible with Pulsar2 version: 6.0 115775d3
17
-
18
- ## Convert tools links:
19
-
20
- For those who are interested in model conversion, you can try to export axmodel through
21
-
22
- - [Pulsar2 Link, How to Convert ONNX to axmodel](https://pulsar2-docs.readthedocs.io/en/latest/pulsar2/introduction.html)
23
-
24
-
25
- ## Support Platform
26
-
27
- - AX650
28
- - [M4N-Dock(爱芯派Pro)](https://wiki.sipeed.com/hardware/zh/maixIV/m4ndock/m4ndock.html)
29
- - [M.2 Accelerator card](https://docs.m5stack.com/en/ai_hardware/LLM-8850_Card)
30
-
31
-
32
- | 模型 | 输入分辨率 | AX650板端耗时 |
33
- |------|-----------|--------------|
34
- | AOD-Net | 640×480 | 2.4 ms |
35
- | Light-Dehazenet | 480×640 | 9.2 ms |
36
- | DehazeFormer_t | 512×512 | 161ms |
37
- | MixDehazeNet | 256×256 | 38 ms |
38
- | GCANet | 512×512 | 80 ms |
39
- | GridDehazeNet | 640×480 | 113 ms |
40
- | DEA-Net | 512×512 | 127 ms |
41
- | FFA-Net | 512×512 | 873 ms |
42
-
43
-
44
- ## How to use
45
-
46
- Download all files from this repository to the device
47
-
48
- ```
49
- 所有模型文件组织方式均如下:
50
- .
51
- |-- model_convert
52
- | |-- axmodel
53
- | | `-- dehazeformer-t-512-constant.axmodel
54
- | `-- dehazeformer.json
55
- |-- pic
56
- | `-- 00000_0_0.1800.png
57
- |-- python
58
- | |-- axmodel_infer.py
59
- | `-- onnx_infer.py
60
- `-- res
61
- `-- output.png
62
-
63
- ```
64
-
65
- ### Inference
66
- #### Inference with AX650 Host, such as M4N-Dock(爱芯派Pro)
67
- 模型推理,执行命令:
68
- ```
69
- (base) root@ax650:~/GCANet# python axmodel_infer.py
70
- [INFO] Available providers: ['AxEngineExecutionProvider', 'AXCLRTExecutionProvider']
71
- [INFO] Using provider: AxEngineExecutionProvider
72
- [INFO] Chip type: ChipType.MC50
73
- [INFO] VNPU type: VNPUType.DISABLED
74
- [INFO] Engine version: 2.12.0s
75
- [INFO] Model type: 2 (triple core)
76
- [INFO] Compiler version: 7.0 22923e4e
77
- Saved: axmodel_output/0051_0.8_0.2_input_dehaze_compare.png
78
- Saved: axmodel_output/0099_0.9_0.16_input_dehaze_compare.png
79
-
80
-
81
- ```
82
-
83
- ```
84
- 推理结果样例:
85
-
86
- ![](GCANet/res/0099_0.9_0.16_input_dehaze_compare.png)
87
- ![](GCANet/res/0051_0.8_0.2_input_dehaze_compare.png)
88
-
89
- ```
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
README.md CHANGED
@@ -1,3 +1,89 @@
1
  ---
2
  license: mit
3
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  ---
2
  license: mit
3
+ language:
4
+ - en
5
+ pipeline_tag: image-to-image
6
+ tags:
7
+ - dehazing
8
+ - ImageEnhancement
9
+
10
+ # ImageDehazing
11
+
12
+ This is a collection of image dehazing algorithms, models have been converted to run on the Axera NPU using **w8a8** quantization.
13
+
14
+ This model has been optimized with the following LoRA:
15
+
16
+ Compatible with Pulsar2 version: 6.0 115775d3
17
+
18
+ ## Convert tools links:
19
+
20
+ For those who are interested in model conversion, you can try to export axmodel through
21
+
22
+ - [Pulsar2 Link, How to Convert ONNX to axmodel](https://pulsar2-docs.readthedocs.io/en/latest/pulsar2/introduction.html)
23
+
24
+
25
+ ## Support Platform
26
+
27
+ - AX650
28
+ - [M4N-Dock(爱芯派Pro)](https://wiki.sipeed.com/hardware/zh/maixIV/m4ndock/m4ndock.html)
29
+ - [M.2 Accelerator card](https://docs.m5stack.com/en/ai_hardware/LLM-8850_Card)
30
+
31
+
32
+ | 模型 | 输入分辨率 | AX650板端耗时 |
33
+ |------|-----------|--------------|
34
+ | AOD-Net | 640×480 | 2.4 ms |
35
+ | Light-Dehazenet | 480×640 | 9.2 ms |
36
+ | DehazeFormer_t | 512×512 | 161ms |
37
+ | MixDehazeNet | 256×256 | 38 ms |
38
+ | GCANet | 512×512 | 80 ms |
39
+ | GridDehazeNet | 640×480 | 113 ms |
40
+ | DEA-Net | 512×512 | 127 ms |
41
+ | FFA-Net | 512×512 | 873 ms |
42
+
43
+
44
+ ## How to use
45
+
46
+ Download all files from this repository to the device
47
+
48
+ ```
49
+ 所有模型文件组织方式均如下:
50
+ .
51
+ |-- model_convert
52
+ | |-- axmodel
53
+ | | `-- dehazeformer-t-512-constant.axmodel
54
+ | `-- dehazeformer.json
55
+ |-- pic
56
+ | `-- 00000_0_0.1800.png
57
+ |-- python
58
+ | |-- axmodel_infer.py
59
+ | `-- onnx_infer.py
60
+ `-- res
61
+ `-- output.png
62
+
63
+ ```
64
+
65
+ ### Inference
66
+ #### Inference with AX650 Host, such as M4N-Dock(爱芯派Pro)
67
+ 模型推理,执行命令:
68
+ ```
69
+ (base) root@ax650:~/GCANet# python axmodel_infer.py
70
+ [INFO] Available providers: ['AxEngineExecutionProvider', 'AXCLRTExecutionProvider']
71
+ [INFO] Using provider: AxEngineExecutionProvider
72
+ [INFO] Chip type: ChipType.MC50
73
+ [INFO] VNPU type: VNPUType.DISABLED
74
+ [INFO] Engine version: 2.12.0s
75
+ [INFO] Model type: 2 (triple core)
76
+ [INFO] Compiler version: 7.0 22923e4e
77
+ Saved: axmodel_output/0051_0.8_0.2_input_dehaze_compare.png
78
+ Saved: axmodel_output/0099_0.9_0.16_input_dehaze_compare.png
79
+
80
+
81
+ ```
82
+
83
+ ```
84
+ 推理结果样例:
85
+
86
+ ![](GCANet/res/0099_0.9_0.16_input_dehaze_compare.png)
87
+ ![](GCANet/res/0051_0.8_0.2_input_dehaze_compare.png)
88
+
89
+ ```