JETSON 人脸检测——LFFD-MNN
文章目录
LFFD-MNN 介绍
可以参考这个博客:人脸识别论文 – CVPR2019-LFFD
源码
参考这个链接:the C++ implemententation of LFFD with MNN
这个是 LFFD-MNN 的代码。
环境搭建及相关的 C++ 部署
因为这次设备 JETSON-ORIN-NX 进行人脸检测主要是为了与后续的硬件融合。
因此主要是使用 C++ 语言。所以需要对源码进行相关的 C++ 部署。
但是本文尝试使用 MNN 进行C++ 部署,成功。因此下面会展示 MNN 部署C++ 并且修改源码为可以对视频进行检测,然后获得结果的过程。
下载并编译 MNN
参考这个链接: Jetson Nano安装MNN深度学习框架
sudo cp ./MNN/build/libMNN.so ./home/be/Desktop/LFFD-MNN-master/MNN/lib
编译 LFFD-MNN
开始进行 LFFD 的编译:
cd LFFD-MNN-master
mkdir build && cd build && cmake ..
make -j$(nproc)
编译完成后可以运行一下测试文件:
./main ../models/ ../data/test.jpg
修改 main.cpp 进行视频读取和人脸检测
修改前的 main.cpp:
#include <iostream>
#include "MNN_LFFD.h"
#include <opencv2/opencv.hpp>
#include <string>
#include <thread>
#include <chrono>
#include <opencv2/highgui/highgui_c.h>
using namespace cv;
#define IMAGE_TEST
//#define VIDEO
// .......
}
修改后的 main.cpp:
#include <iostream>
#include "MNN_LFFD.h"
#include <opencv2/opencv.hpp>
#include <string>
#include <thread>
#include <chrono>
#include <opencv2/highgui/highgui_c.h>
using namespace cv;
//#define IMAGE_TEST
#define VIDEO
// .......
}
然后在终端输入
cd LFFD-MNN-master/build
make -j4
推荐的视频下载网站:视觉中国旗下网站(https://www.vcg.com/creative-video/)
下载好视频后, data 文件夹下,然后重命名为 test.mp4。
运行下面的测试命令即可进行测试:
./main ../models/ ../data/test.mp4
CPU 资源占用等数据的获取
此时通过在终端中输入 top:
即可获得代码运行过程中占用资源的数据。
其中 CPU 占用量最大的就是当前的人脸检测代码。