模型训练与测试----Loss的角度

 print('Epoch: [{0}][{1}/{2}]\t'
              'Lr: {lr:.6f}\t'
              'Time {batch_time.val:.2f} ({batch_time.avg:.2f})\t'
              'Data {data_time.val:.4f} ({data_time.avg:.4f})\t'
              'Loss {loss.val:.4f} ({loss.avg:.4f})\t'.format(
            epoch + 1, batch_idx + 1, len(trainloader), batch_time=batch_time,
            data_time=data_time, loss=losses, lr=optimizer.param_groups[0]['lr'],
            ))  #rank1=rank1, mAP=mAP

代码中有写到多少batch回输出loss的以及各种,其中.val 是方差的计算,而avg是均值的计算。所以可以通过每次的输出看出这么多个batch的损失函数的变化是多少,而且可以从方差看出,这么多个batch的损失输出是不是都稳定在这个范围。通过这样的判断可以看出损失输出的判断,从而判断模型的学习情况。比如:我用了OCR的156层的输出来计算学习情况:

Epoch: [1][60/554]	Lr: 0.001000	Time 0.67 (0.64)	Data 0.0002 (0.0100)	Loss 0.1253 (0.2844)	
Epoch: [1][120/554]	Lr: 0.001000	Time 0.54 (0.62)	Data 0.0001 (0.0051)	Loss 0.1507 (0.2735)	
Epoch
  • 0
    点赞
  • 6
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
基于YOLO的人脸识别模型是一种基于深度学习的目标检测算法,它可以实现实时的人脸检测和识别。YOLO(You Only Look Once)是一种单阶段的目标检测算法,相比于传统的两阶段方法,它具有更快的检测速度和更高的准确率。 训练一个基于YOLO的人脸识别模型需要以下步骤: 1. 数据收集:首先需要收集包含人脸的图像数据集。这些图像可以来自于公开的数据集,也可以通过自己采集。确保数据集中包含各种不同角度、光照条件和表情的人脸图像。 2. 标注数据:对收集到的图像进行标注,即给每个图像中的人脸框出一个矩形框,并标注其类别为人脸。可以使用标注工具如LabelImg来完成这个过程。 3. 数据预处理:对标注好的数据进行预处理,包括图像尺寸调整、数据增强等操作。数据增强可以通过随机裁剪、旋转、翻转等方式增加数据样本的多样性。 4. 构建模型:选择合适的YOLO模型架构,如YOLOv3或YOLOv4,并根据数据集的特点进行调整。可以使用深度学习框架如TensorFlow或PyTorch来构建模型。 5. 模型训练:使用标注好的数据集对模型进行训练训练过程中需要定义损失函数,常用的是YOLO的损失函数YOLO Loss。通过反向传播算法不断优化模型参数,直到模型收敛。 6. 模型评估:使用测试集对训练好的模型进行评估,计算模型在人脸检测和识别任务上的准确率、召回率等指标。 7. 模型应用:将训练好的模型应用于实际场景中,进行人脸检测和识别任务。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值