yolo利用txt可视化碰到的问题

问题一:

IndexError: too many indices for array: array is 1-dimensional, but 2 were indexed

报错原因:

可视化用的txt文件夹里面有的txt文件是0kb,无法可视化,程序报错。

参考地址:

遇到IndexError: too many indices for array: array is 1-dimensional, but 2 were i..问题_dear_queen的博客-CSDN博客icon-default.png?t=M3K6https://blog.csdn.net/dear_queen/article/details/116230861?ops_request_misc=%257B%2522request%255Fid%2522%253A%2522165061835316782184641455%2522%252C%2522scm%2522%253A%252220140713.130102334.pc%255Fall.%2522%257D&request_id=165061835316782184641455&biz_id=0&utm_medium=distribute.pc_search_result.none-task-blog-2~all~first_rank_ecpm_v1~rank_v31_ecpm-4-116230861.142^v9^pc_search_result_cache,157^v4^control&utm_term=IndexError%3A+too+many+indices+for+array%3A+array+is+1-dimensional%2C+but+2+were+indexed&spm=1018.2226.3001.4187

解决办法:

将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输出改成如上图所示就好了。

你可能感兴趣的:(python,深度学习)