首页 > 其他分享 >yolov8训练过程中,出现IndexError:index 17 is out of bounds for dimension 1 with size 17,如何解决问题

yolov8训练过程中,出现IndexError:index 17 is out of bounds for dimension 1 with size 17,如何解决问题

时间:2024-06-24 09:57:35浏览次数:22  
标签:8.1 index py IndexError 17 self ultralytics File line

  在用yolov8做数据训练自己的数据时发现,这样一个错误,困扰了我很久。报错的原因是数组的问题,我查了一下百度,说是定义数组的问题,之后我就慌的一批,这个源包这么多,该去哪排查。

raceback (most recent call last):
  File "d:\jiaotong\ultralytics-8.1.0\my train.py", line 10, in <module>
    results = model.train(data='yolo-bvn.yaml', epochs=20, imgsz=640)
  File "d:\jiaotong\ultralytics-8.1.0\ultralytics\engine\model.py", line 390, in train
    self.trainer.train()
  File "d:\jiaotong\ultralytics-8.1.0\ultralytics\engine\trainer.py", line 208, in train
    self._do_train(world_size)
  File "d:\jiaotong\ultralytics-8.1.0\ultralytics\engine\trainer.py", line 379, in _do_train
    self.loss, self.loss_items = self.model(batch)
  File "C:\Users\15200\.conda\envs\ultralytics\lib\site-packages\torch\nn\modules\module.py", line 1532, in _wrapped_call_impl
    return self._call_impl(*args, **kwargs)
  File "C:\Users\15200\.conda\envs\ultralytics\lib\site-packages\torch\nn\modules\module.py", line 1541, in _call_impl
    return forward_call(*args, **kwargs)
  File "d:\jiaotong\ultralytics-8.1.0\ultralytics\nn\tasks.py", line 79, in forward
    return self.loss(x, *args, **kwargs)
  File "d:\jiaotong\ultralytics-8.1.0\ultralytics\nn\tasks.py", line 258, in loss  
    return self.criterion(preds, batch)
  File "d:\jiaotong\ultralytics-8.1.0\ultralytics\utils\loss.py", line 220, in __call__
    _, target_bboxes, target_scores, fg_mask, _ = self.assigner(
  File "C:\Users\15200\.conda\envs\ultralytics\lib\site-packages\torch\nn\modules\module.py", line 1532, in _wrapped_call_impl
    return self._call_impl(*args, **kwargs)
  File "C:\Users\15200\.conda\envs\ultralytics\lib\site-packages\torch\nn\modules\module.py", line 1541, in _call_impl
    return forward_call(*args, **kwargs)
  File "C:\Users\15200\.conda\envs\ultralytics\lib\site-packages\torch\utils\_contextlib.py", line 115, in decorate_context
    return func(*args, **kwargs)
  File "d:\jiaotong\ultralytics-8.1.0\ultralytics\utils\tal.py", line 72, in forward
    mask_pos, align_metric, overlaps = self.get_pos_mask(
  File "d:\jiaotong\ultralytics-8.1.0\ultralytics\utils\tal.py", line 94, in get_pos_mask
    align_metric, overlaps = self.get_box_metrics(pd_scores, pd_bboxes, gt_labels, gt_bboxes, mask_in_gts * mask_gt)
  File "d:\jiaotong\ultralytics-8.1.0\ultralytics\utils\tal.py", line 113, in get_box_metrics
    bbox_scores[mask_gt] = pd_scores[ind[0], :, ind[1]][mask_gt]  # b, max_num_obj, h*w
IndexError: index 17 is out of bounds for dimension 1 with size 17

经过排查,终于找到了,一个十分低级的错误,我在修改的coco8.yaml文件时,待识别的类个数少了一个。这个是我们在使用labelimg时,标注图片类的个数,一一对应打上去。这两个要一摸一样。

之后在运行代码,进行训练就可以,下一期会出一个完整版的使用yolov8进行训练自己的数据集。

标签:8.1,index,py,IndexError,17,self,ultralytics,File,line
From: https://blog.csdn.net/weixin_60819517/article/details/139915123

相关文章

  • 【译】VisualStudio.Extensibility 17.10:用 Diagnostics Explorer 调试您的扩展
    想象一下,创建的扩展比以往任何时候都运行得更快、更流畅!如果您最近还没有跟上,我们一直在努力改进VisualStudio.ExtensibilitySDK。VisualStudio.Extensibility帮助您构建在主IDE进程之外运行的扩展,以提高性能和可靠性。它还提供了一个时尚而直观的基于.NET8的API......
  • 5034. 【NOI2017模拟3.28】B —— 矩阵树定理和拉格朗日插值的结合
    题目大意给你一棵\(n\)(\(n\le50\))个点的树,可以进行不超过\(k\)次操作,每次断掉一条边,再连上一条边,要求树一直是树,求一共有多少种树的形态。思路把题意转换为对于一个\(n\)个点的完全图,是树边的话权值是\(1\),否则是\(x\)。跑一遍矩阵树定理,矩阵树定理求的是一个图所有生......
  • 【TWVRP】遗传算法求解带时间窗的载重约束外卖配送车辆路径规划问题【含Matlab源码 14
    ......
  • 17-oop封装
    封装信息隐藏:应禁止直接访问一个对象中数据的实际表示,而应通过接口操作来访问属性私有,get/set追求“高内聚,低耦合”//类//封装的核心关键字:private私有publicclassStudent{//属性私有privateStringname;//名字privateintid;//学号private......
  • CF1707E Replace
    题目描述给定一个长为\(n\)的序列\(a_1,\ldots,a_n\),其中对于任意的\(i\)满足\(1\leqa_i\leqn\)。定义一个二元组函数如下:\[f((l,r))=(\min\{a_l,\ldots,a_r\},\max\{a_l,\ldots,a_r\})(l\leqr)\]你需要回答\(q\)次询问,每次给定\((l_i,r_i)\),问其最少经过多少......
  • CF1770E Koxia and Tree
    题目描述给定一棵\(n\)个点的树,在\(k\)个位置上存在蝴蝶,我们需要给\(n-1\)条边定向,如果一条边的起点有蝴蝶且终点没有蝴蝶,那么蝴蝶将被移动到终点,我们会按照给定边的顺序移动,问最终所有蝴蝶的树上距离的和的期望,答案除于\(\frac{k(k-1)}{2}\),对\(998244353\)取模\[k\len\le300......
  • 胡说八道(24.6.17)——STM32以及通信杂谈
        之前的文章中咱们谈到了STM32的时钟,今天我们来联系实际,来看看内部时钟下和外部时钟下的两种不同时钟的电平翻转。本次终于有硬件了,是最基础的STM32F103C8T6。    首先是,内部时钟的配置操作。             系统的内部时钟是72MHz,由上图......
  • 力扣-1705. 吃苹果的最大数目
    1.题目介绍题目地址(1705.吃苹果的最大数目-力扣(LeetCode))https://leetcode.cn/problems/maximum-number-of-eaten-apples/题目描述有一棵特殊的苹果树,一连n天,每天都可以长出若干个苹果。在第i天,树上会长出apples[i]个苹果,这些苹果将会在days[i]天后(也就是说,第i+......
  • JDK17新增语法特征
    1、引言从springboot3.0开始,已经不支持JDK8了。从3.0开始,转变为JDK17。 选用 Java17,概括起来主要有下面几个主要原因:1. JDK17是LTS(长期支持版),可以免费商用到2029年。而且将前面几个过渡版(JDK9-JDK16)去其糟粕,取其精华的版本;2......
  • elasticsearch-7.17.15 集群安装部署及kibana配置
    一、物料准备(注意:必须版本一致):1、安装包 elasticsearch-7.17.15-linux-x86_64.tar.gz(这个版本的插件需要在线使用命令安装:/es/elasticsearch-7.17.15/bin/elasticsearch-plugininstallhttps://get.infini.cloud/elasticsearch/analysis-ik/7.17.15,或者用我的传送门) an......