代码:
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
posted on 2022-03-09 18:06  一点飞鸿  阅读(3274)  评论(0编辑  收藏  举报