IndexError: index 54 is out of bounds for axis 1 with size 3错误解决
本人小白一个,部分术语可能有误,在这将错误记录一下,以供参考。
用yolov5训练自己的数据集时,运行yolov5的test.py文件报错。原因是没有将权重参数改为训练好的权重。
本人错误代码如下:
这里使用的是yolov5自带的yolov5s的预训练权重,将它改为训练好的权重即可。
用yolov5训练自己的数据集时,运行yolov5的test.py文件报错。原因是没有将权重参数改为训练好的权重。
本人错误代码如下:
这里使用的是yolov5自带的yolov5s的预训练权重,将它改为训练好的权重即可。