这个是在比特大陆BM1684系列AI硬件(SC5+/SE5/SM5)上运行RetinaFace人脸检测模型的demo
-
前提
已有了bmnnsdk2-bm1684_v2.2.0的sdk,并且安装完lib。
-
使用说明
a) cd bmnnsdk2-bm1684_v2.2.0/examples
b) git clone https://github.com/AnBaolei1984/BMRetinaFaceDemo.git
c) cd BMRetinaFaceDemo
d) 编译程序
如果是SC5+板卡 : make -f Makefile.pcie -j4 如果是SE5或SM5 : make -f Makefile.arm -j4
e) 运行程序
如果是图片模式, ./face_test 0 imagelist.txt face_detection_fp32.bmodel 如果是视频模式, ./face_test 1 videolist.txt face_detection_int8_b4.bmodel * 1 imagelist.txt的每一行是图片的路径 * 2 videolist.txt的每一行是一个mp4视频路径或者一个rtsp url * 3 如果模型是多batch的,那么图片模式,会每攒够batch数的图片做一次推理。视频模式, 要求videolist.txt的视频数和模型的batch数相等。当然也可以按着自己需求的逻辑来 修改。只是demo里这样写的。
f) 结束后会在当前目录生成一个名为result_imgs的文件夹,里面可以看到结果图片