今天介绍的是一个使用yolox目标检测算法,实现换人脸的过程。
首先是换人脸视频的准备,我们可以找一些比较的热门视频作为素材,比如“华强买瓜”。
使用视频抽帧,将一个2分钟的视频,抽成200张左右,当然也可以抽成更多张。
以上就是视频抽帧的结果。
然后我们下载labelimg工具,使用python 在命令行输入pip install labelimg即可。
下载好labelimg之后,对数据进行打标签操作,每个人脸做上相应的logo,由于这次使用初级的人脸替换,只用一个标签face即可,后续会使用更多表情,未来做到更加逼真的人脸替换。
在对每人脸打上label标签face之后。我们做成了VOC的数据集格式。
再在网上找个现成开源的目标检测代码对数据集进行训练模型即可
最终训练完成之后,我们进行人脸框的检测,检测结果像这张图一样,一个个人脸被检测出来了。
可以看到人脸已经被很好的检测出来了。
然后我们再根据坐标替换上我们像替换上的搞笑图片即可。
最后把这个目标检测的框去掉就可以了
完美实现换脸。后续会不断优化这一实现方法,实现更多功能。



