代码:
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   一点飞鸿  阅读(3427)  评论(0编辑  收藏  举报
努力加载评论中...
点击右上角即可分享
微信分享提示