基于多任务卷积网络(MTCNN)和Center-Loss的多人实时人脸检测和人脸识别系统


**DFace** 是个开源的深度学习人脸检测和人脸识别系统。所有功能都采用 **[pytorch](https://github.com/pytorch/pytorch)** 框架开发。pytorch是一个由facebook开发的深度学习框架,它包含了一些比较有趣的高级特性,例如自动求导,动态构图等。DFace天然的继承了这些优点,使得它的训练过程可以更加简单方便,并且实现的代码可以更加清晰易懂。 DFace可以利用CUDA来支持GPU加速模式。我们建议尝试linux GPU这种模式,它几乎可以实现实时的效果。
资源截图
代码片段和文件信息
import cv2
from dface.core.detect import create_mtcnn_net MtcnnDetector
from dface.core import vision


if __name__ == ‘__main__‘:

    # refer to your local model path 
    p_model = “./model_store/pnet_epoch.pt“
    r_model = “./model_store/rnet_epoch.pt“
    o_model = “./model_store/onet_epoch.pt“

    #use cpu version set use_cuda=False if you want to use gpu version set use_cuda=True
    pnet rnet onet = create_mtcnn_net(p_model_path=p_model r_model_path=r_model o_model_path=o_model use_cuda=False)
    mtcnn_detector = MtcnnDetector(pnet=pnet rnet=rnet onet=onet min_face_size=24)

    img = cv2.imread(“./test.jpg“)
    b g r = cv2.split(img)
    img2 = cv2.merge([r g b])

    bboxs landmarks = mtcnn_detector.detect_face(img)
    # print box_align

    vision.vis_face(img2bboxslandmarks)

 属性            大小     日期    时间   名称
----------- ---------  ---------- -----  ----
     目录           0  2018-03-28 17:56  dface
     文件       10250  2018-03-28 17:56  dfaceLICENSE
     文件        8055  2018-03-28 17:56  dfaceREADME.md
     目录           0  2018-03-28 17:56  dfaceanno_store
     文件           0  2018-03-28 17:56  dfaceanno_store\__init__.py
     文件          55  2018-03-28 17:56  dfaceanno_storeinfo
     文件     3208713  2018-03-28 17:56  dfaceanno_storewider_origin_anno.txt
     目录           0  2018-03-28 17:56  dfacedface
     文件           0  2018-03-28 17:56  dfacedface\__init__.py
     文件        1037  2018-03-28 17:56  dfacedfaceconfig.py
     目录           0  2018-03-28 17:56  dfacedfacecore
     文件           0  2018-03-28 17:56  dfacedfacecore\__init__.py
     文件       22527  2018-03-28 17:56  dfacedfacecoredetect.py
     文件        4224  2018-03-28 17:56  dfacedfacecoreimage_reader.py
     文件        1180  2018-03-28 17:56  dfacedfacecoreimage_tools.py
     文件        5310  2018-03-28 17:56  dfacedfacecoreimagedb.py
     文件        6848  2018-03-28 17:56  dfacedfacecoremodels.py
     文件        1150  2018-03-28 17:56  dfacedfacecore
ms.py
     文件        8950  2018-03-28 17:56  dfacedfacecore
esnet_inception_v2.py
     文件          20  2018-03-28 17:56  dfacedfacecore
oc.py
     文件        2653  2018-03-28 17:56  dfacedfacecoreutils.py
     文件        5136  2018-03-28 17:56  dfacedfacecorevision.py
     目录           0  2018-03-28 17:56  dfacedfaceprepare_data
     文件           0  2018-03-28 17:56  dfacedfaceprepare_data\__init__.py
     文件         979  2018-03-28 17:56  dfacedfaceprepare_dataassemble.py
     文件         984  2018-03-28 17:56  dfacedfaceprepare_dataassemble_onet_imglist.py
     文件         996  2018-03-28 17:56  dfacedfaceprepare_dataassemble_pnet_imglist.py
     文件         996  2018-03-28 17:56  dfacedfaceprepare_dataassemble_rnet_imglist.py
     文件        8007  2018-03-28 17:56  dfacedfaceprepare_datagen_Onet_train_data.py
     文件        6709  2018-03-28 17:56  dfacedfaceprepare_datagen_Pnet_train_data.py
     文件        7696  2018-03-28 17:56  dfacedfaceprepare_datagen_Rnet_train_data.py
............此处省略30个文件信息

版权声明:本文内容由互联网用户自发贡献,该文观点仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌抄袭侵权/违法违规的内容, 请发送邮件举报,一经查实,本站将立刻删除。

发表评论

评论列表(条)