【代码笔记】计算点云中各类别的点数

count_sum = np.zeros((13,1), dtype=np.int)
scene = 'data/s3dis/pointcnn'
for area in ['Area_1', 'Area_2', 'Area_3', 'Area_4', 'Area_6']:
    for room in os.listdir(os.path.join(scene, area)):
        print(os.path.join(scene, area, room, 'label.npy'))
        ground_truth = np.load(os.path.join(scene, area, room, 'label.npy'))
        uni_label, count = np.unique(ground_truth, return_counts=True)
        for i in range(len(uni_label)):
            count_sum[int(uni_label[i])] += count[i]
print(count_sum)