• 大小: 3.71MB
    文件类型: .zip
    金币: 1
    下载: 0 次
    发布日期: 2023-11-17
  • 语言: 其他
  • 标签: 人脸识别  

资源简介

**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  dface\LICENSE
     文件        8055  2018-03-28 17:56  dface\README.md
     目录           0  2018-03-28 17:56  dface\anno_store\
     文件           0  2018-03-28 17:56  dface\anno_store\__init__.py
     文件          55  2018-03-28 17:56  dface\anno_store\info
     文件     3208713  2018-03-28 17:56  dface\anno_store\wider_origin_anno.txt
     目录           0  2018-03-28 17:56  dface\dface\
     文件           0  2018-03-28 17:56  dface\dface\__init__.py
     文件        1037  2018-03-28 17:56  dface\dface\config.py
     目录           0  2018-03-28 17:56  dface\dface\core\
     文件           0  2018-03-28 17:56  dface\dface\core\__init__.py
     文件       22527  2018-03-28 17:56  dface\dface\core\detect.py
     文件        4224  2018-03-28 17:56  dface\dface\core\image_reader.py
     文件        1180  2018-03-28 17:56  dface\dface\core\image_tools.py
     文件        5310  2018-03-28 17:56  dface\dface\core\imagedb.py
     文件        6848  2018-03-28 17:56  dface\dface\core\models.py
     文件        1150  2018-03-28 17:56  dface\dface\core\nms.py
     文件        8950  2018-03-28 17:56  dface\dface\core\resnet_inception_v2.py
     文件          20  2018-03-28 17:56  dface\dface\core\roc.py
     文件        2653  2018-03-28 17:56  dface\dface\core\utils.py
     文件        5136  2018-03-28 17:56  dface\dface\core\vision.py
     目录           0  2018-03-28 17:56  dface\dface\prepare_data\
     文件           0  2018-03-28 17:56  dface\dface\prepare_data\__init__.py
     文件         979  2018-03-28 17:56  dface\dface\prepare_data\assemble.py
     文件         984  2018-03-28 17:56  dface\dface\prepare_data\assemble_onet_imglist.py
     文件         996  2018-03-28 17:56  dface\dface\prepare_data\assemble_pnet_imglist.py
     文件         996  2018-03-28 17:56  dface\dface\prepare_data\assemble_rnet_imglist.py
     文件        8007  2018-03-28 17:56  dface\dface\prepare_data\gen_Onet_train_data.py
     文件        6709  2018-03-28 17:56  dface\dface\prepare_data\gen_Pnet_train_data.py
     文件        7696  2018-03-28 17:56  dface\dface\prepare_data\gen_Rnet_train_data.py
............此处省略30个文件信息

评论

共有 条评论