YOLOV5代码理解——损失函数的计算
YOLOV5代码理解——损失函数的计算摘要: 神经网络的训练的主要流程包括图像输入神经网络, 得到模型的输出结果,计算模型的输出与真实值的损失, 计算损失值的梯度,最后用梯度下降算法更新模型参数。损失函数值的计算是非常关键的一个步骤。本博客将对yolov5损失值的计算过程代码的实现做简要的理解。def compute_loss(p, targets, model): # predictions, targets, model device = targets.device lcls,
原创
2021-06-29 23:42:06 ·
23124 阅读 ·
10 评论