在上一节我们学习了人脸检测,但是它只能检测出那是一张人脸,无法识别那个人脸是谁,今天我们就来学习如何识别人脸,识别有没有戴口罩,识别交通标志。
其实在博派上面的用法和之前的20种物体识别以及人脸检测是一模一样的,不同的只是锚点参数和模型文件,我们先来看看效果。
以口罩检测,没戴口罩显示为Un_Masks,戴了口罩显示为Masks,效果如下:
直接上代码
其实更改的只是这几个参数,其他代码都是一样,那这些参数是怎么来的呢?
我们用YOLO训练完一个模型后,一般会得到如下几个文件:
我们分别打开两个文本文件看看里面的内容
这里面就是我们需要设置的参数,把这些参数复制下来粘贴到代码里即可。
人脸识别也是如此,我做了一个模型识别刘德华和郭富城,效果如下:
还有交通标志的识别也是如此:
代码都是一样,只是参数和模型文件不同,就不贴图了,大家直接下载模型文件和代码自己尝试。
记住一定要先把模型文件复制到TF卡里哦。
评论