基于多任务卷积网络(MTCNN)和Center-Loss的多人实时人脸检测和人脸识别系统
**DFace** 是个开源的深度学习人脸检测和人脸识别系统。所有功能都采用 **[pytorch](https://github.com/pytorch/pytorch)** 框架开发。pytorch是一个由facebook开发的深度学习框架,它包含了一些比较有趣的高级特性,例如自动求导,动态构图等。DFace天然的继承了这些优点,使得它的训练过程可以更加简单方便,并且实现的代码可以更加清晰易懂。
DFace可以利用CUDA来支持GPU加速模式。我们建议尝试linux GPU这种模式,它几乎可以实现实时的效果。
代码片段和文件信息
属性 大小 日期 时间 名称
----------- --------- ---------- ----- ----
目录 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个文件信息
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个文件信息
版权声明:本文内容由互联网用户自发贡献,该文观点仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌抄袭侵权/违法违规的内容, 请发送邮件举报,一经查实,本站将立刻删除。
评论列表(条)