DBface

最新推荐文章于 2024-04-25 09:50:23 发布

wi162yyxq

最新推荐文章于 2024-04-25 09:50:23 发布

阅读量894

点赞数 1

分类专栏：深度学习 python

本文链接：https://blog.csdn.net/wi162yyxq/article/details/106354337

版权

python 同时被 2 个专栏收录

25 篇文章 1 订阅

订阅专栏

深度学习

13 篇文章 1 订阅

订阅专栏

网络其实还是蛮简单的，多尺度进行人脸检索，最厉害的地方就是模型只有7MB，速度很快，我的破电脑只有不到2G显存都能实时检测。

使用方法也极其简单

def image_demo():

    dbface = DBFace()
    dbface.eval()

    if HAS_CUDA:
        dbface.cuda()

    dbface.load("model/dbfaceSmallH.pth")
    detect_image(dbface, "datas/selfie.jpg")
    detect_image(dbface, "datas/12_Group_Group_12_Group_Group_12_728.jpg")


def camera_demo():

    dbface = DBFace()
    dbface.eval()

    if HAS_CUDA:
        dbface.cuda()

    dbface.load("model/dbfaceSmallH.pth")
    cap = cv2.VideoCapture(0)
    cap.set(cv2.CAP_PROP_FRAME_WIDTH, 1080)
    cap.set(cv2.CAP_PROP_FRAME_HEIGHT, 720)
    ok, frame = cap.read()

    while ok:
        objs = detect(dbface, frame)

        for obj in objs:
            common.drawbbox(frame, obj)

        cv2.imshow("demo DBFace", frame)
        key = cv2.waitKey(1) & 0xFF
        if key == ord('q'):
            break

        ok, frame = cap.read()
    
    cap.release()
    cv2.destroyAllWindows()