代碼:
cate_ids=np.unique(gt_box_array[:,-1])
for tmp_cateid in cate_ids:
conf_matrix[tmp_cateid,0]+=1
原因:numpy里面不指定類型的話,默認是float64位,無法作為索引
修改:強轉為int ,即conf_matrix[int(tmp_cateid),0]+=1