Update README.md

README.md

@@ -10,14 +10,12 @@ base_model:
 pipeline_tag: depth-estimation
 
 tags:
+- Axera
 - RAFT-Stereo
+- NPU
 - Stereo-Matching
-- Disparity-Estimation
 - Depth-Estimation
 - Computer-Vision
-- 3DV-2021
-- PyTorch
-- Convolutional-GRU
 ---
 
 # RAFT-Stereo on Axera NPU
@@ -44,21 +42,56 @@ For model conversion and deployment guidance:
   - [LLM630 Compute Kit](https://docs.m5stack.com/zh_CN/core/LLM630%20Compute%20Kit)
 
 
-| Chips | Model Variant | NPU3 Latency (ms) |
+| Chips | Model Variant | Latency (ms) |
 | --- | --- | --- |
-| AX650 | RAFT-Stereo-Small | TBD |
+| AX650 | raft_steoro256x640_r1.axmodel | 20.9 |
+| AX650 | raft_steoro256x640_r4.axmodel | 111.8 |
 
-| Chips | Model Variant | NPU2 Latency (ms) |
+| Chips | Model Variant | Latency (ms) |
 | --- | --- | --- |
-| AX630C| RAFT-Stereo-Small | TBD |
+| AX630C| raft_steoro256x640_r1_npu2.axmodel | 317.765 |
+| AX630C| raft_steoro256x640_r4_npu2.axmodel | 825.793 |
 
 ## How to Use
 
 Download the repository and ensure the directory structure is organized as follows:
 
 ```text
-
-
+├── ax630c
+│   ├── raft_steoro256x640_r1.onnx
+│   ├── raft_steoro256x640_r1_npu2.axmodel
+│   ├── raft_steoro256x640_r4.onnx
+│   └── raft_steoro256x640_r4_npu2.axmodel
+├── ax650
+│   ├── raft_steoro256x640_r1.axmodel
+│   ├── raft_steoro256x640_r1.onnx
+│   ├── raft_steoro384x1280_r4.axmodel
+│   └── raft_steoro384x1280_r4.onnx
+├── examples
+│   ├── left
+│   │   ├── 000051_11.png
+│   │   ├── 000058_11.png
+│   │   ├── 000059_10.png
+│   │   ├── 000121_10.png
+│   │   ├── 000164_10.png
+│   │   ├── 000167_11.png
+│   │   ├── 000172_11.png
+│   │   ├── 000179_10.png
+│   │   ├── 000193_10.png
+│   │   └── 000195_10.png
+│   └── right
+│       ├── 000051_11.png
+│       ├── 000058_11.png
+│       ├── 000059_10.png
+│       ├── 000121_10.png
+│       ├── 000164_10.png
+│       ├── 000167_11.png
+│       ├── 000172_11.png
+│       ├── 000179_10.png
+│       ├── 000193_10.png
+│       └── 000195_10.png
+├── infer.py
+└── infer_onnx.py
 ```
 
 ### Prerequisites
@@ -86,7 +119,7 @@ python3 infer.py --left examples/left/000051_11.png --right examples/right/00005
 The model generates a dense disparity map representing the depth of the scene. Results are saved as colorized depth maps and point cloud projections where applicable.
 
 ```
-(base) root@ax630c:~/data# python3 infer.py --left examples/left/000051_11.png --right examples/right/000051_11.png --model raft_steoro256x640_r1_npu2.axmodel --width 640 --height 256
+(base) root@ax630c:~/data# /python3 infer.py --left examples/left/000051_11.png --right examples/right/000051_11.png --model raft_steoro256x640_r4_npu2.axmodel --width 640 --height 256
 [INFO] Available providers:  ['AxEngineExecutionProvider']
 [INFO] Using provider: AxEngineExecutionProvider
 [INFO] Chip type: ChipType.MC20E