用modelnet40_ply_hdf5_2048数据集制作txt点云数据集（抽取特定类别）

import os
import numpy as np
import h5py
BASE_DIR=os.path.dirname(os.path.abspath(__file__))

def getDataFiles(list_filename):
    return [line.rstrip() for line in open(list_filename)]

def load_h5(h5_filename):
    f = h5py.File(h5_filename)
    data = f['data'][:]
    label = f['label'][:]
    normal=f['normal'][:]
    return (data, label , normal)

def loadDataFile(filename):
    return load_h5(filename)

TRAIN_FILES = getDataFiles( \
    os.path.join(BASE_DIR, 'modelnet40_ply_hdf5_2048/train_files.txt'))
TEST_FILES = getDataFiles(\
    os.path.join(BASE_DIR, 'modelnet40_ply_hdf5_2048/test_files.txt'))
a=[]#保存每个类别的数据
b=[]#测试集类别数据
c=[]#保存家居点云
d=[]#测试集家具点云
for i in range(40):
    a.append([])
for i in range(40):
    b.append([])
# for i in range(14):
#     c.append([])
# for i in range(14):
#     d.append([])
for fn in range(len(TRAIN_FILES)):
    current_data,current_label,current_normal&#