记录错误:IndexError: index 5 is out of bounds for axis 1 with size 3

最近在跑代码,错误多的忘记记录了。
错误记录:IndexError: index 5 is out of bounds for axis 1 with size 3
原因:这是在使用YOLOv3进行木目标检测的时候遇到的错误,原因时权重放错了,导致内部的维度出现了偏差。
解决方法:用该数据训练的权重进行相应的测试相应数据即可。

你可能感兴趣的:(目标检测,python,pytorch,神经网络,机器学习,深度学习)