Yolov6解决常见报错（1）TypeError numpy.float64 object cannot be interpreted as an index

原创

于 2022-07-16 13:23:56 发布 · 1.7k 阅读

7 ·

CC 4.0 BY-SA版权

文章标签：

#python

本文介绍了在尝试运行Yolov6时遇到的TypeError问题，详细解析了报错原因，指出在numpy 1.11.0之后的版本中，浮点数不能作为索引。提供了解决方案，即在pycocotools/cocoeval.py文件的setDetParams和setKpParams函数中，用numpy.arange替换np.linspace来解决此问题。同时感谢schmiloo的文章提供帮助。

这几天偶然看见Yolov6出来，迫不及待的试了一下，结果看见网上评论说bug太多了，我作为使用者，想着积极做出贡献，把一些我解决的bug分享一下，方便大家也能看到顺便解决。

先看报错

Training completed in 0.082 hours.
Traceback (most recent call last):
File "tools/train.py", line 92, in
main(args)
File "tools/train.py", line 82, in main
trainer.train()
File "/mnt/batch/tasks/shared/LS_root/mounts/clusters/multi-gpu-4-tesla-m60/code/Users/zhanghe/cv-safety-detection/YOLOv6/yolov6/core/engine.py", line 70, in train
self.train_in_loop()
File "/mnt/batch/tasks/shared/LS_root/mounts/clusters/multi-gpu-4-tesla-m60/code/Users/zhanghe/cv-safety-detection/YOLOv6/yolov6/core/engine.py", line 89, in train_in_loop
self.eval_and_save()
File "/mnt/batch/tasks/shared/LS_root/mounts/clusters/multi-gpu-4-tesla-m60/code/Users/zhanghe/cv-safety-detection/YOLOv6/yolov6/core/engine.py", line 115, in eval_and_save
self.eval_model()
File "/mnt/batch/tasks/shared/LS_root/mounts/clusters/multi-gpu-4-tesla-m60/code/Users/zhanghe/cv-safety-detection/Y