waterman 发表于 2024-12-4 02:48

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
```
之后等待编译完成即可。

# 程序测试
编译完成后会在 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
```

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

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

- 在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
```


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

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

Jacktang 发表于 2024-12-4 07:31

<p>期待后面进一步实现rtsp推流+多人人脸识别的测评</p>
页: [1]
查看完整版本: AI挑战营(进阶):4. 板端Retinaface+facenet测试