首页 > 其他分享 >使用yolo训练时Loss变为nan[解决思路]

使用yolo训练时Loss变为nan[解决思路]

时间:2023-01-25 14:32:17浏览次数:60  
标签:Loss yolo 函数 检查 random nan 不当


首先参考这篇文章:

​原创|使用caffe训练时Loss变为nan的原因​​:

  • 梯度爆炸
  • 不当的损失函数
  • 不当的输入
  • 池化层中步长比核的尺寸大

检查自己的train.py:

  1. 检查代码(正确)
  2. 检查输入(自己之前调试的时候修改了图片预处理的一个函数get_random_data)将random=False改成random=True后就可以了

标签:Loss,yolo,函数,检查,random,nan,不当
From: https://blog.51cto.com/u_12630471/6022623

相关文章

  • yolov7 tensorrt模型加速部署【实战】
    0.linux环境配置基于tensorrt+cudac++实现模型end2end的gpu加速,支持win10、linux,在2023年已经更新模型:YOLOv8,YOLOv7,YOLOv6,YOLOv5,YOLOv4,YOLOv3,YOLOX,YOLOR,......
  • 【实战】yolov8 tensorrt模型加速部署
    【实战】yolov8tensorrt模型加速部署TensorRT-Alpha基于tensorrt+cudac++实现模型end2end的gpu加速,支持win10、linux,在2023年已经更新模型:YOLOv8,YOLOv7,YOLOv6,YOLO......
  • save() prohibited to prevent data loss due to unsaved related object 'item_n
    问题描述:save()prohibitedtopreventdatalossduetounsavedrelatedobject'item_no原因分析:原来的目的是保存数据时同步另一个表,不存在新增,存在更新,但......
  • #0029. 「JOI Open Contest 2021」Financial Report
    碎碎念1:是的时隔两年多笨人又想开始更博客了碎碎念2:另外今年就要AFO了希望能给自己的oi生涯画上一个完美的句号!题目大意给定\(N\)个数字和\(D\)需要从中选择一些数字......
  • javaScript中为什么字符串相减是NAN Why is “string” - “string” NaN?
    发现这个问题的背景是后端返回数据根据组排序,然后每个组中的对象按照某个字段进行排序显示一开始是这样的  并不生效修改后    排序生效。可以看到只......
  • YOLOv8 初体验
    简介YOLOv8模型设计快速,准确,易于使用,使其成为广泛的目标检测和图像分割任务的绝佳选择。TheYOLOv8modelisdesignedtobefast,accurate,andeasytouse,making......
  • 【多任务学习】Multi-Task Learning Using Uncertainty to Weigh Losses for Scene Ge
    ·阅读摘要:  本文提出针对CV领域的多任务模型,设置一个可以学习损失权重的损失层,可以提高模型精度。·参考文献:  [1]Multi-TaskLearningUsingUncertaintytoWeig......
  • Yolov5_v6.2训练数据集进行预测
    Yolov5_v6.2训练数据集进行预测学习笔记记录一下第一次进行Yolov5的部署,调试,训练,预测。第一次发布在了CSDN:链接:https://blog.csdn.net/C_1Y1_T/article/details/12769......
  • YOLO家族系列模型的演变:从v1到v8(下)
    昨天的文章中,我们回顾了YOLO家族的前9个架构。本文中将继续总结最后3个框架,还有本月最新发布的YOLOV8.Backbone最初由一个分支(GoogLeNet、VGG、Darknet)组成,然后过......
  • yolov5
    参考网址:https://zhuanlan.zhihu.com/p/501798155 源码下载及使用release下载source及pt文件(yolov5s.pt)https://github.com/ultralytics/yolov5/tagshttps://github.co......