1.配置文件中Optimizer参数为auto的时候,Optimizer会怎么选用呢?
当迭代次数大于10000时则会使用SGD,否则使用AdamW。
关于迭代的计算:dataset:数据集中图片的数量。 nbs:默认为64。 batchsize、epoch:可自己设置。
iterations=dataset/max(nbs,batchsize)*epoch。
训练中的进度条为:dataset/batchsize
ps:建议直接选择使用SGD
2.best.pt根据什么指标来保存的?
def fitness(self):
"""Model fitness as a weighted combination of metrics."""
w = [0.0, 0.0, 0.1, 0.9] # weights for [P, R, mAP@0.5, mAP@0.5:0.95]
return (np.array(self.mean_results()) * w).sum()
根据0.1*map0.5+0.9*map0.5:0.95来保存best.pt.
3.预测框问题
3.1预测框的线的宽度
利用default.yaml中的line_width进行修改,默认线的宽度为2。
3.2类别、置信度等显示
3.3输出精度
修改ultralytics\models\yolo\detect\val.py中的
pf = '%22s' + '%11i' * 2 + '%11.3g' * len(self.metrics.keys) # print format
0.3表示显示小数点后三位。
3.4预测框的颜色修改
修改ultralytics\utils\plotting.py中的hex中的颜色代码。