Update README.md
Browse files
README.md
CHANGED
|
@@ -44,9 +44,11 @@ Download the repository and ensure the directory structure is organized as follo
|
|
| 44 |
.
|
| 45 |
βββ centerpoint.axmodel # The compiled Axera model
|
| 46 |
βββ inference_axmodel.py # Main inference script
|
| 47 |
-
βββ
|
| 48 |
-
βββ config
|
| 49 |
-
|
|
|
|
|
|
|
| 50 |
|
| 51 |
|
| 52 |
### Prerequisites
|
|
@@ -62,7 +64,7 @@ Download the repository and ensure the directory structure is organized as follo
|
|
| 62 |
Run the inference script by providing the compiled model, configuration, and data directory.
|
| 63 |
|
| 64 |
```bash
|
| 65 |
-
python inference_axmodel.py centerpoint.axmodel
|
| 66 |
|
| 67 |
```
|
| 68 |
|
|
@@ -70,6 +72,20 @@ python inference_axmodel.py centerpoint.axmodel inference_data/config/inference_
|
|
| 70 |
|
| 71 |
### Results
|
| 72 |
The model generates a 3D detection map with bounding boxes oriented in 3D space. Results are saved as images and videos which visualize the ego-vehicle, point cloud data, and detected objects.
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 73 |
|
| 74 |
### Example Visualization
|
| 75 |
|
|
|
|
| 44 |
.
|
| 45 |
βββ centerpoint.axmodel # The compiled Axera model
|
| 46 |
βββ inference_axmodel.py # Main inference script
|
| 47 |
+
βββ extracted_data/ # Input directory
|
| 48 |
+
βββ config.json # Configuration files (e.g., inference_config.json)
|
| 49 |
+
βββ sample_index.json
|
| 50 |
+
βββ gt_annotations/
|
| 51 |
+
βββ points/
|
| 52 |
|
| 53 |
|
| 54 |
### Prerequisites
|
|
|
|
| 64 |
Run the inference script by providing the compiled model, configuration, and data directory.
|
| 65 |
|
| 66 |
```bash
|
| 67 |
+
python inference_axmodel.py ./centerpoint.axmodel ./extracted_data/config.json ./extracted_data --output-dir ./inference_results --visualize --num-samples 50 --score-thr 0.5
|
| 68 |
|
| 69 |
```
|
| 70 |
|
|
|
|
| 72 |
|
| 73 |
### Results
|
| 74 |
The model generates a 3D detection map with bounding boxes oriented in 3D space. Results are saved as images and videos which visualize the ego-vehicle, point cloud data, and detected objects.
|
| 75 |
+
```
|
| 76 |
+
(ax_env) root@ax650:~/data# python inference_axmodel.py ./centerpoint.axmodel ./extracted_data/config.json ./extracted_data --output-dir ./inference_results --visualize --num-samples 50 --score-thr 0.5
|
| 77 |
+
[INFO] Available providers: ['AxEngineExecutionProvider']
|
| 78 |
+
[INFO] Using provider: AxEngineExecutionProvider
|
| 79 |
+
[INFO] Chip type: ChipType.MC50
|
| 80 |
+
[INFO] VNPU type: VNPUType.DISABLED
|
| 81 |
+
[INFO] Engine version: 2.12.0s
|
| 82 |
+
[INFO] Model type: 2 (triple core)
|
| 83 |
+
[INFO] Compiler version: 5.1-patch1 ed388aa0
|
| 84 |
+
Processing 50 samples...
|
| 85 |
+
Inference: 100%|ββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββ| 50/50 [00:47<00:00, 1.06it/s]
|
| 86 |
+
Creating video: 100%|βββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββ| 50/50 [00:02<00:00, 23.32it/s]
|
| 87 |
+
Done! 50 frames, 12836 detections, saved to ./inference_results
|
| 88 |
+
```
|
| 89 |
|
| 90 |
### Example Visualization
|
| 91 |
|