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)
12-01
1876
10-12
277
12-27
1415