问题一:
IndexError: too many indices for array: array is 1-dimensional, but 2 were indexed
报错原因:
可视化用的txt文件夹里面有的txt文件是0kb,无法可视化,程序报错。
参考地址:
解决办法:
将0kb的txt文件删除,同时将删除的txt文件名相对应的照片也删除就可以了。
问题二:
照片根据txt可视化完成以后发现可视化的照片上面无任何标签框。
在csdn找了好久没没有找到相关的问题,就对程序进行了单步调试,发现我的txt文件里面前四位是pred_boxes,第五位是pred_labels,但是在我读取的时候是读取的第一位是标签,所以在检测生成txt的时候改变一下输出的顺序就好了。先输出pred_labels,后输出pred_boxs.
print( int(pred_labels[i]), *[int(p) for p in pred_boxes[i]], float(pred_scores[i]), file=f)
txt输出改成如上图所示就好了。