逛了逛腾讯的github,确实开源了不少好东西,这不就觉得FaceDetection-DSFD看起来检测人脸很好用
项目地址是https://github.com/Tencent/FaceDetection-DSFD
年代较为久远,最近更新也是在15个月之前了,框架还是Torch1.0
但是年代久远不意味着它不满足我们的需求,没有最好的模型,只有最佳的解决方案。
然后尝试了一番发现它跑不起来。。。
代码虽然写的很清楚,但是各种小bug,语焉不详的配置项,各种包的版本兼容问题很恼火
issue一查大家都是,这不好心人就出现了:
https://github.com/vlad3996/FaceDetection-DSFD
首先我们没有那么大的数据集支撑我们从新训练,就算重新训练了精度也并不一定比预训练权重的更好,就算比预训练的更好也不一定比他的泛化能力更强
其次里面很多代码都是为训练用了,推理很多都用不到
这个老哥就把没用的玩意都干掉了,留下的精简了一下
import cv2
import torch
fro