免费注册 查看新帖 |

ChinaUnix.net

  平台 论坛 博客 文库 频道自动化运维 虚拟化 储存备份 C/C++ PHP MySQL 嵌入式 Linux系统
最近访问板块 发新帖
查看: 2152 | 回复: 0

怎么计算脸部姿势 with cv2.solvePnP in python3 [复制链接]

论坛徽章:
0
发表于 2018-03-21 10:38 |显示全部楼层
我用shape_predictor_68_face_landmarks.dat计算出脸部得68个点,想得到脸部姿势,通过这个函数计算在两个人得图片上计算,发现[float(roll), float(pitch), float(yaw)]值接近得两张图,脸得姿势并不一样,谁用过这个函数,帮忙给点意见。
def face_orientation(frame, landmarks):
    size = frame.shape #(height, width, color_channel)

    # image_points = np.array([
                            # (landmarks[4], landmarks[5]),     # Nose tip
                            # (landmarks[10], landmarks[11]),   # Chin
                            # (landmarks[0], landmarks[1]),     # Left eye left corner
                            # (landmarks[2], landmarks[3]),     # Right eye right corne
                            # (landmarks[6], landmarks[7]),     # Left Mouth corner
                            # (landmarks[8], landmarks[9])      # Right mouth corner
                        # ], dtype="double")
   
    # center_righteyes = (landmarks[36] + landmarks[39]) // 2
    # center_lefteyes  = (landmarks[42] + landmarks[45]) // 2
    # image_points = numpy.array([
                            # landmarks[31],     # Nose tip
                            # landmarks[8],   # Chin
                            # center_lefteyes,     # Left eye left corner
                            # center_righteyes,     # Right eye right corne
                            # landmarks[48],     # Left Mouth corner
                            # landmarks[54]      # Right mouth corner
                        # ], dtype="double")
                        
    image_points = numpy.array([
        (landmarks[33, 0], landmarks[33, 1]),     # Nose tip
        (landmarks[8, 0], landmarks[8, 1]),       # Chin
        (landmarks[36, 0], landmarks[36, 1]),     # Left eye left corner
        (landmarks[45, 0], landmarks[45, 1]),     # Right eye right corner
        (landmarks[48, 0], landmarks[48, 1]),     # Left Mouth corner
        (landmarks[54, 0], landmarks[54, 1])      # Right mouth corner
        ], dtype="double")

                        
    model_points = numpy.array([
                            (0.0, 0.0, 0.0),             # Nose tip
                            (0.0, -330.0, -65.0),        # Chin
                            (-165.0, 170.0, -135.0),     # Left eye left corner
                            (165.0, 170.0, -135.0),      # Right eye right corne
                            (-150.0, -150.0, -125.0),    # Left Mouth corner
                            (150.0, -150.0, -125.0)      # Right mouth corner                        
                        ])

    # Camera internals

    center = (size[1]/2, size[0]/2)
    focal_length = center[0] / numpy.tan(60/2 * numpy.pi / 180)
    camera_matrix = numpy.array(
                         [[focal_length, 0, center[0]],
                         [0, focal_length, center[1]],
                         [0, 0, 1]], dtype = "double"
                         )

    dist_coeffs = numpy.zeros((4,1)) # Assuming no lens distortion
    #(success, rotation_vector, translation_vector) = cv2.solvePnP(model_points, image_points, camera_matrix, dist_coeffs, flags=cv2.CV_ITERATIVE)
    (success, rotation_vector, translation_vector) = cv2.solvePnP(model_points, image_points, camera_matrix, dist_coeffs, flags=cv2.SOLVEPNP_ITERATIVE)

   
    axis = numpy.float32([[500,0,0],
                          [0,500,0],
                          [0,0,500]])
                          
    imgpts, jac = cv2.projectPoints(axis, rotation_vector, translation_vector, camera_matrix, dist_coeffs)
    modelpts, jac2 = cv2.projectPoints(model_points, rotation_vector, translation_vector, camera_matrix, dist_coeffs)
    rvec_matrix = cv2.Rodrigues(rotation_vector)[0]

    proj_matrix = numpy.hstack((rvec_matrix, translation_vector))
    eulerAngles = cv2.decomposeProjectionMatrix(proj_matrix)[6]

   
    pitch, yaw, roll = [math.radians(_) for _ in eulerAngles]


    pitch = math.degrees(math.asin(math.sin(pitch)))
    roll = -math.degrees(math.asin(math.sin(roll)))
    yaw = math.degrees(math.asin(math.sin(yaw)))

    #return imgpts, modelpts, (str(int(roll)), str(int(pitch)), str(int(yaw))), (landmarks[4], landmarks[5])
    return numpy.array([float(roll), float(pitch), float(yaw)], dtype = "double")
您需要登录后才可以回帖 登录 | 注册

本版积分规则 发表回复

数据风云,十年变迁
DTCC 第十届中国数据库技术大会已启航!

2019年5月8日~5月10日,由IT168旗下ITPUB企业社区平台主办的第十届中国数据库技术大会(DTCC2019),将在北京隆重召开。大会将邀请百余位行业专家,就热点技术话题进行分享,是广大数据领域从业人士的又一次年度盛会和交流平台。与SACC2018类似,本届大会将采用“3+2”模式:3天传统技术演讲+2天深度主题培训。大会不仅提供超100场的主题演讲,还会提供连续2天的深度课程培训,深化数据领域的项目落地实践方案。
DTCC2019,一场值得期待的数据技术盛会,殷切地希望您报名参与!

活动入口>>
  

北京盛拓优讯信息技术有限公司. 版权所有 16024965号-6 北京市公安局海淀分局网监中心备案编号:11010802020122
中国互联网协会会员  联系我们:huangweiwei@it168.com
感谢所有关心和支持过ChinaUnix的朋友们 转载本站内容请注明原作者名及出处

清除 Cookies - ChinaUnix - Archiver - WAP - TOP