zoukankan      html  css  js  c++  java
  • tensorflow+图像读取+人脸特征点检测+三维顶点保存

    import numpy as np
    import os
    from glob import glob
    import scipy.io as sio
    from skimage.io import imread, imsave
    from time import time
    
    from api import PRN
    from utils.write import write_obj
    
    # ---- init PRN
    os.environ['CUDA_VISIBLE_DEVICES'] = '0' # GPU number, -1 for CPU
    prn = PRN(is_dlib = False) 
    
    
    # ------------- load data
    image_folder = 'TestImages/AFLW2000/'
    save_folder = 'TestImages/AFLW2000_results'
    if not os.path.exists(save_folder):
        os.mkdir(save_folder)
    
    types = ('*.jpg', '*.png')
    image_path_list= []
    for files in types:
        image_path_list.extend(glob(os.path.join(image_folder, files)))
    total_num = len(image_path_list)
    
    for i, image_path in enumerate(image_path_list):
        # read image
        image = imread(image_path)
    
        # the core: regress position map    
        if 'AFLW2000' in image_path:
            mat_path = image_path.replace('jpg', 'mat')
            info = sio.loadmat(mat_path)
            kpt = info['pt3d_68']
            pos = prn.process(image, kpt) # kpt information is only used for detecting face and cropping image
        else:
            pos = prn.process(image) # use dlib to detect face
    
        # -- Basic Applications
        # get landmarks
        kpt = prn.get_landmarks(pos)
        # 3D vertices
        vertices = prn.get_vertices(pos)
        # corresponding colors
        colors = prn.get_colors(image, vertices)
    
        # -- save
        name = image_path.strip().split('/')[-1][:-4]
        np.savetxt(os.path.join(save_folder, name + '.txt'), kpt) 
        write_obj(os.path.join(save_folder, name + '.obj'), vertices, colors, prn.triangles) #save 3d face(can open with meshlab)
    
        sio.savemat(os.path.join(save_folder, name + '_mesh.mat'), {'vertices': vertices, 'colors': colors, 'triangles': prn.triangles})
  • 相关阅读:
    noi放苹果
    二分 网线主管
    hdu 1421 dp
    hdu 1087 最大上升子序列的和(dp或线段树)
    快速排序+查找
    zoj 1425 最大交叉匹配
    hdu 3501 容斥原理或欧拉函数
    hdu 4671 异面直线的距离
    hdu 3320 计算几何(三维图形几何变换)
    hdu 2857 点在直线上的投影+直线的交点
  • 原文地址:https://www.cnblogs.com/hxjbc/p/10510783.html
Copyright © 2011-2022 走看看