AI挑战营（进阶）:4. 板端Retinaface+facenet测试

waterman

AI挑战营（进阶）:4. 板端Retinaface+facenet测试 [复制链接]

在上一节中我们得到了retinaface和facenet的两个rknn模型，在本节我们将以 https://github.com/LuckfoxTECH/luckfox_pico_rknn_example.git 库中的luckfox_pico_retinaface_facenet_demo为例，快速验证我们得到的模型能否正确推理。
# 程序编译
首先将库克隆下来，
```
git clone https://github.com/LuckfoxTECH/luckfox_pico_rknn_example.git
```
之后设置环境变量，把LUCKFOX_SDK_PATH设置为我们SDK的目录
```
export LUCKFOX_SDK_PATH=/home/sdk/luckfox-pico
```

获取仓库源码并设置自动编译脚本执行权限
```
chmod a+x ./build.sh
./build.sh
```
执行 ./build.sh 后选择编译的例程
```
1) luckfox_pico_retinaface_facenet
2) luckfox_pico_retinaface_facenet_spidev
3) luckfox_pico_yolov5
Enter your choice [1-3]:1
```
之后等待编译完成即可。

# 程序测试
编译完成后会在 install 文件夹下生成对应的**部署文件夹**，我们这里使用到的是luckfox_pico_retinaface_facenet_demo。
```
luckfox_pico_retinaface_facenet_demo
luckfox_pico_retinaface_facenet_spidev_pro_max_demo
luckfox_pico_retinaface_facenet_spidev_plus_demo
luckfox_pico_yolov5_demo
```

<ignore_js_op>

<a href="javascript:;" onclick="showWindow(this.id, this.getAttribute('url'), 'get', 0);" id="savephoto_867107" url="home.php?mod=spacecp&ac=album&op=saveforumphoto&aid=867107&handlekey=savephoto_867107">保存到相册</a>

</p>

</div>
<div class="tip_horn"></div>
</div>

</ignore_js_op>

将luckfox_pico_retinaface_facenet_demo/model目录中的两个模型文件换成我们自己训练得到的rknn文件。

<ignore_js_op>

<a href="javascript:;" onclick="showWindow(this.id, this.getAttribute('url'), 'get', 0);" id="savephoto_867108" url="home.php?mod=spacecp&ac=album&op=saveforumphoto&aid=867108&handlekey=savephoto_867108">保存到相册</a>

</p>

</div>
<div class="tip_horn"></div>
</div>

</ignore_js_op>

将生成的luckfox_pico_retinaface_facenet_demo目录使用adb完整上传到 Luckfox上
```
adb push ./luckfox_pico_retinaface_facenet_demo /
```

<ignore_js_op>

<a href="javascript:;" onclick="showWindow(this.id, this.getAttribute('url'), 'get', 0);" id="savephoto_867109" url="home.php?mod=spacecp&ac=album&op=saveforumphoto&aid=867109&handlekey=savephoto_867109">保存到相册</a>

</p>

</div>
<div class="tip_horn"></div>
</div>

</ignore_js_op>

- 在Luckfox板端中赋予其中的可执行程序执行权限
```
cd /luckfox_pico_retinaface_facenet_demo
chmod 777 luckfox_pico_retinaface_facenet
```
执行luckfox_pico_retinaface_facenet
```
./luckfox_pico_retinaface_facenet ./model/retinaface.rknn ./model/facenet.rknn ./model/test.jpg
```

<ignore_js_op>

<a href="javascript:;" onclick="showWindow(this.id, this.getAttribute('url'), 'get', 0);" id="savephoto_867110" url="home.php?mod=spacecp&ac=album&op=saveforumphoto&aid=867110&handlekey=savephoto_867110">保存到相册</a>

</p>

</div>
<div class="tip_horn"></div>
</div>

</ignore_js_op>

可以看到，当检测到人脸时，会将人脸的锚框位置和与输入人脸图片的距离输出至命令行。由于输入人脸图片为憨豆先生，测试时摄像头采集的人脸为我自己的脸，因此可以看到人脸距离还是挺大的。

以上说明我们的模型基本是没有问题的，能够正常执行推理。接下来就能够进一步实现rtsp推流+多人人脸识别了。

Jacktang

期待后面进一步实现rtsp推流+多人人脸识别的测评

AI挑战营（进阶）:4. 板端Retinaface+facenet测试 [复制链接]

最新回复

浏览过的版块