PlytoObj（python“.ply”文件转“.obj”格式）

import osimport numpy as npfrom glob import globfrom plyfile import PlyDatafrom tqdm import tqdmdef write_obj(verts, faces, obj_path): """ Write .obj file """ assert obj_path[-4:] == '.obj' with open(obj_path, 'w') as fp: for v in verts: fp.write('v %f %f %fn' % (v[0], v[1], v[2])) for f in faces+1: fp.write('f %d %d %dn' % (f[0], f[1], f[2]))def ply2obj(ply_name, obj_name): plydata = PlyData.read(ply_name) pc = plydata['vertex'].data # print(pc) faces = plydata['face'].data # print(faces) try: pc_array = np.array([[x, y, z] for x,y,z,_,_,_ in pc]) except: pc_array = np.array([[x, y, z] for x,y,z in pc]) face_array = np.array([face[0] for face in faces], dtype=np.int) write_obj(pc_array, face_array, obj_name)if __name__ == "__main__": filelist = glob(r'路径*.ply') # print(filelist) for i in tqdm(range(len(filelist))): if os.path.exists(filelist[i].replace('.ply','.obj')): continue ply2obj(filelist[i], filelist[i].replace('.ply','.obj')) #目标文件与源文件保存在同一目录