深度学习(四):自己训练yolov5模型进行目标检测_怎么用训练好的yolo模型检测图片-CSDN博客
kmeans的源码我看的是这个博客。
代码现在还在跑,以下的错误我就不贴图片了,之后有需要的话再修改贴图。
1.路径报错
刚刚开始跑的时候报了一个位置路径的错,我本来以为是绝对路径和相对路径的问题,也搜索了ubuntu绝对路径的格式,用gedit也可以进,所以不是路径的问题,后来发现clauculate_anchors.py里面的这个文件夹路径不是txt格式的文件夹路径,是xml格式的,所以考虑格式转换。
python 批量txt转xml_txt转成xml python-CSDN博客
格式转换用的这个博客的源码
2.a must be greater than 0 unless no samples are taken
根据报错,我的问题行如图所示,第一张图是clauculate_anchors.py,第二张图是kmeans.py
打印了很多值之后我发现问题主要在于下面这个for循环之后得出的cls_name是数字,和我的数据集名字class_name对不上,所以进不去下面的if, 最后这个函数的返回值是[],没有具体的值导致报错,修改class_name为,class_name = ['0','1'](标签数),没有报错。
如果有不对的地方或者存疑的地方希望大家指正。