yolov8修改检测框
时间: 2023-11-28 16:06:16 浏览: 314
yolov8所有的检测模型都可以转onnx代码
根据提供的引用内容,yolov8修改检测框的方法如下:
1. 在训练集中随机挑选200张左右作为训练集,加快验证速度;
2. 将box loss的权重扩大为原来的2倍,训练30个epoch左右,观察验证集上的检测框是否正常;
3. 如果检测框依然过小,尝试将初始学习率扩大为原来的2倍、4倍、5倍,训练30个epoch左右,观察验证集上的检测框是否正常;
4. 重复步骤2和步骤3直至检测框正常;
5. 更换为原始数据集,如果调大batchsize, 初始学习率也要相应增大。
如果1调整后检测框依然过小,可以按照以下方式调整:
1. 找到所能检出框的最大边长公式:(self.reg_max-1)*2*max_stride;
2. 原始yolov8中,strides为[8, 16, 32],self.reg_max为16,因此检测框的最大边长为(16-1)x2x32=960;
3. 根据实际情况调整self.reg_max和max_stride的值,以增大检测框的最大边长。
阅读全文