This shows you the differences between two versions of the page.
Both sides previous revision Previous revision Next revision | Previous revision | ||
products:sbc:vim4:npu:demos:retinaface [2023/09/18 01:44] sravan |
products:sbc:vim4:npu:demos:retinaface [2024/01/29 02:16] (current) louis |
||
---|---|---|---|
Line 5: | Line 5: | ||
{{indexmenu_n> | {{indexmenu_n> | ||
===== Get source code ===== | ===== Get source code ===== | ||
+ | |||
+ | [[gh> | ||
```shell | ```shell | ||
Line 32: | Line 34: | ||
Follow Docker official documentation to install Docker: [[https:// | Follow Docker official documentation to install Docker: [[https:// | ||
- | Then fetch the prebuilt NPU Docker | + | Then fetch the prebuilt NPU Docker |
```shell | ```shell | ||
Line 52: | Line 54: | ||
==== Convert ==== | ==== Convert ==== | ||
- | After training the model, we should convert the PyTorch model to ONNX model. Create the python | + | After training the model, we should convert the PyTorch model into an ONNX model. Create the Python |
```python export.py | ```python export.py | ||
Line 98: | Line 100: | ||
``` | ``` | ||
- | Run '' | + | Run '' |
```shell | ```shell | ||
Line 152: | Line 154: | ||
# Run | # Run | ||
- | $ sudo ./ | + | $ sudo ./ |
``` | ``` | ||
- | '' | + | '' |