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})
  • 相关阅读:
    datatable linq查询操作filed用法
    C# asp.net 中sql like in 参数化
    c#.net全站防止SQL注入类的代码
    js面向对象编程:if中可以使用那些作为判断条件呢?
    HTTP Method 详细解读(`GET` `HEAD` `POST` `OPTIONS` `PUT` `DELETE` `TRACE` `CONNECT`)--转
    浅析HTTP代理原理--转
    细说Cookie--转
    HTTP协议7之Cookie--转
    HTTP协议6之状态码--转
    HTTP协议5之代理--转
  • 原文地址:https://www.cnblogs.com/hxjbc/p/10510783.html
Copyright © 2011-2022 走看看