https://github.com/deepinsight/insightface
github上insightnFace地址
insightFace中的RetinaFace的使用,2020年2月24日前后发布了口罩检测流程,在文件夹RetinaFaceAntiCov中。
使用RetinaFaceAntiCov文件夹内容需要进行如下修改:
1. 复制 RetinaFace中rcnn文件夹下的cython文件夹到RetinaFaceAntiCov的rcnn文件夹中,在rcnn->processing的bbox_transform.py中,修改import 如下图所示:红框中的内容为本地路径,注意修改为自己的路径。
2. 进行如下修改,在rcnn->cython 的cpu_nms.pyx(不是图中的nms.pyx),将25行的np.int_t 修改为 np.intp_t
3. 在命令行中转入RetinaFaceAntiCov\rcnn\cython 文件夹内,执行命令:
python setup.py build_ext --inplace
4. 在RetinaFaceAntiCov 下执行test.py文件,修改模型路径和需要识别的图片路径以及保存图片的路径
模型路径,/mnet_cov2前面为下图两个文件的地址,可以在百度云盘:下载
https://pan.baidu.com/s/16ihzPxjTObdbv0D6P6LmEQ j3b6
需要识别的图片
保存识别之后的图片的路径