ModelNet40数据集h5文件加载及说明

数据集介绍

ModelNet40数据集是一个点云分类数据集,训练集9840个点云,测试集2468个点云,共40个类别
ModelNet40数据集h5文件加载及说明_第1张图片
json里是每个训练/测试集里各个点云的名称在这里插入图片描述
h5文件存的是点云数据,加载出来看看格式

all_data = []
all_label = []
for h5_name in glob.glob(os.path.join(DATA_DIR, 'modelnet40_ply_hdf5_2048', 'ply_data_%s*.h5'%"train")):
    f = h5py.File(h5_name)
    data = f['data'][:].astype('float32')
    label = f['label'][:].astype('int64')
    f.close()
    all_data.append(data)
    all_label.append(label)
all_data = np.concatenate(all_data, axis=0)
all_label = np.concatenate(all_label, axis=0)

ModelNet40数据集h5文件加载及说明_第2张图片
在这里插入图片描述

可以看到训练集有9840个点云,每个点云有2048个点,包含每个点的坐标xyz,每个点云有一个类别标签,同理可查看测试集中的点云个数为2468
ModelNet40数据集h5文件加载及说明_第3张图片

你可能感兴趣的:(分类,深度学习,人工智能)