如何将 HDF5 数据集中的 3D 点云数据导出为 .ply 文件格式

Question

给定一个 HDF5 文件，其下具有 HDF5 数据集类型的 complete_pcds，如下所示。

这个 complete_pcds 有点像一个数组，其中每一行都包含一个 3D 模型的数据。我想将此 complete_pcds 的每一行转换为 .ply 文件，目的是最终将其视为 3D 模型。包括 complete_pcds 文件的属性以供参考。

Answer 1

这是我试图回答的问题。我对您的数据和架构仍然有些模糊。 您的屏幕截图显示(2400, 2048, 3) 的形状。据此，我假设您需要 2400 个输出文件，每个文件有 2048 个顶点，其中 x,y,x 坐标是数据集 [file#,vertex#,0:2] 中的值。此外，您的文件没有任何元素面定义，因此不会创建它们。如果这是正确的，请继续阅读……如果不是，请更正/澄清我的假设。

另外，我创建了一个简单的 HDF5 进行测试。创建它的代码在答案的末尾。该示例有 4 行数据；每行有 8 个点来定义立方体的角（每行中的立方体都被缩放 - 第一行是 1x1x1，第二行是 2x2x2，依此类推）。

以下是我用于 ply 格式的参考：

• PLY - Polygon File Format
• File Formats - PLY file

将数据提取到 .ply 文件的代码：

import h5py
filename = 'MVP_Train_CP_ex.h5'
with h5py.File(filename,'r') as h5f: 
    ply_ds = h5f['complete_pcds']
    print(ply_ds.shape, ply_ds.dtype)
    for cnt, row in enumerate(ply_ds):
        v_cnt = row.shape[0]
        with open(f'pcds_{str(cnt)}.ply','w') as ply_f:
            ply_f.write('ply\n')
            ply_f.write('format ascii 1.0\n')
            ply_f.write(f'comment row {cnt} exported from:{filename}\n')
            ply_f.write(f'element vertex {v_cnt}\n') # variable # of vertices
            ply_f.write('property float x\n')
            ply_f.write('property float y\n')
            ply_f.write('property float z\n')
            ply_f.write('end_header\n')
            for vertex in row:
                ply_f.write(f'{vertex[0]:#6.3g} {vertex[1]:#6.3g} {vertex[2]:#6.3g}\n')

创建示例文件的代码：

import numpy as np
import h5py 
filename = 'MVP_Train_CP_ex.h5'
with h5py.File(filename,'w') as h5f: 
    vertices = np.empty(shape=(4,8,3))
    for i in range(vertices.shape[0]):
        vertices[i,:,:] = (i+1)* np.array([ \
                    [ 0.0, 0.0, 0.0],
                    [ 1.0, 0.0, 0.0],
                    [ 1.0, 1.0, 0.0],
                    [ 0.0, 1.0, 0.0],
                    [ 0.0, 0.0, 1.0],
                    [ 1.0, 0.0, 1.0],
                    [ 1.0, 1.0, 1.0],
                    [ 0.0, 1.0, 1.0]])    
    
    ds = h5f.create_dataset('complete_pcds',data=vertices)