qq_66048450 2023-04-23 21:55 采纳率: 33.3%
浏览 90
已结题

yolov5训练模型出问题

YOLOv5-6.1训练一个四分类模型时出现如下错误，请问大家如何修改

  File "F:\yolo\road_detect\yolov5-v6.1\utils\loss.py", line 148, in __call__
   t[range(n), tcls[i]] = self.cp
IndexError: index 9 is out of bounds for dimension 1 with size 4

img

  • 写回答

2条回答 默认 最新

  • 爱晚乏客游 2023-04-24 09:22
    关注

    分类请运行yolov5/classify/train.py这个文件，而不是yolov5/train.py,这两个是不一样的

    本回答被题主选为最佳回答 , 对您是否有帮助呢?
    评论
    1人已打赏
查看更多回答(1条)

报告相同问题？

  • YOLOv5训练模型的大小问题 opencv python 目标检测
    2023-04-02 10:22
    回答 3 已采纳 该回答通过自己思路及引用到各个渠道搜索综合及思考,得到内容具体如下： 训练出来的不同模型的best.pt文件大小差异很大可能有以下几个原因： 1、模型结构的不同。YOLOv5支持多个不同大小的模型，
  • yolov5训练模型时遇到nan 人工智能 机器学习 算法
    2022-12-31 20:12
    回答 2 已采纳 很遗憾，我无法确定你遇到的问题的原因。在没有更多的信息的情况下，我只能给出一些潜在的建议。 首先，如果你发现模型使用更大的权重文件时在 GPU 上运行会导致数据全部变为 nan，可能是因为你的 GP
  • yolov5 训练模型报错 python 人工智能 有问必答 深度学习
    2022-02-19 21:51
    回答 2 已采纳 Train Custom Data · ultralytics/yolov5 Wiki · GitHub YOLOv5 🚀 in PyT
  • YOLOv5预训练模型权重
    2020-12-09 17:29
    同时，由于YOLOv5模型采用了统一的接口，因此与各种编程语言（如Python）和框架（如PyTorch）的兼容性良好，便于集成到现有的项目中。 总之，YOLOv5预训练模型权重为开发者提供了一套强大且灵活的目标检测工具，...
  • YOLOv5训练以后没办法识别 python 机器学习 目标检测
    2023-04-18 15:26
    回答 3 已采纳 最后发现是分辨率的问题
  • Kaggle训练Yolov5模型 python 有问必答
    2022-04-08 20:06
    回答 2 已采纳 指令输进去,类似于python train.py --.yaml --cfg .yaml --weights pretrained/yolov5s.pt --epoch 100 --batch-siz
  • yolov5模型训练问题 pytorch 机器学习 目标检测
    2022-10-11 21:09
    回答 3 已采纳 你的pytorch的cuda版本选错了，torch1.7.0是支持cuda11.x的，但是你下载的是cu101,这个就表示支持的cuda是10.1。但是你环境里面的cuda是11.x（30系显卡不支持
  • yolov5训练数据集，红细胞检测模型
    2022-07-04 14:49
    Python作为主要的编程语言，用于编写数据处理和模型训练的代码。深度学习是整个项目的核心技术，通过神经网络学习从图像中提取特征，实现红细胞的精确检测。 综上所述，本项目涉及的关键知识点包括YOLOv5模型架构、...
  • yolov5训练损失为Nan python
    2022-08-12 13:39
    回答 1 已采纳 将学习率改为0.00001，步子太大，扯到蛋了，改小点
  • yolov5训练问题 python 人工智能 深度学习
    2023-03-24 16:29
    回答 2 已采纳 如果你想增加数据集数量，然后重新从头开始训练YOLOv5模型，那么建议使用预训练的权重文件yolov5s.pt。这是因为预训练的权重文件yolov5s.pt包含了YOLOv5模型在COCO数据集上的预
  • yolov5中的训练问题 pytorch 深度学习 计算机视觉
    2022-04-27 17:05
    回答 1 已采纳 蓝色框都是配置文件，红色 参数应该是训练集类别的名字，也就是第0类有帮助望采纳，谢谢1！
  • yolov5封装成dll
    2022-06-11 15:24
    同时，需要下载YOLOv5的源代码，并使用`torchscript`将训练好的模型转换为可执行的 TorchScript 模型，便于C++接口使用。 2. **编写C++接口**：创建`.cpp`和`.h`文件，定义函数以加载YOLOv5模型，接收图像数据，...
  • 基于yolov5的RGBDIR四通道茶叶嫩芽检测模型
    2023-12-26 13:17
    Python作为实现该模型的主要编程语言，是因为Python拥有强大的数据处理和科学计算库，如NumPy、Pandas和Matplotlib，以及深度学习库如TensorFlow和PyTorch。YOLOv5就是在PyTorch框架下实现的，PyTorch以其动态计算图...
  • yolo编程YOLOv5 jetson nano 工地防护检测
    2024-08-06 19:27
    环境准备 Jetson Nano 设置 确保你的Jetson Nano已经安装了JetPack...性能优化：Jetson Nano性能有限，确保使用较小的YOLOv5模型（如YOLOv5s）并优化推理过程。 电源管理：使用高性能模式（MAXN模式）来获得最佳性能：
  • yolov8模型的训练方法内含详细步骤可参考
    2024-02-21 21:59
    - **安装Python和PyTorch**：Python是主要的编程语言，而PyTorch则是深度学习框架之一，支持动态图计算，非常适合构建复杂的神经网络。 - **安装ultralytics等库**：ultralytics是YOLOv8的主要开发库，提供了大量的...
  • 没有解决我的问题, 去提问

问题事件

  • 系统已结题 5月2日
  • 已采纳回答 4月24日
  • 创建了问题 4月23日

悬赏问题

  • ¥15 宇视监控服务器无法登录
  • ¥15 PADS Logic 原理图
  • ¥15 PADS Logic 图标
  • ¥15 电脑和power bi环境都是英文如何将日期层次结构转换成英文
  • ¥15 DruidDataSource一直closing
  • ¥20 气象站点数据求取中~
  • ¥15 如何获取APP内弹出的网址链接
  • ¥15 wifi 图标不见了 不知道怎么办 上不了网 变成小地球了
  • ¥50 STM32单片机传感器读取错误
  • ¥50 power BI 从Mysql服务器导入数据，但连接进去后显示表无数据