IndexError: index 2 is out of bounds for dimension 1 with size 2【yolov5自训练报错日志及解决】

yolov5

yolov5 - Ultralytics YOLOv8的前身，是一个用于目标检测、图像分割和图像分类任务的先进模型。

项目地址：https://gitcode.com/gh_mirrors/yo/yolov5

免费下载资源

owenzi0

2914人浏览 · 2023-07-17 22:30:57

owenzi0 · 2023-07-17 22:30:57 发布

Q：自训练到17%时报错：

IndexError: index 2 is out of bounds for dimension 1 with size 2

S：检查出测试集中标签发现残留2号标签。确认该2号标签应为1号标签类别。将2号标签修改为1号标签。（本数据集只分为0和1标签两类，不含2）。

GitHub 加速计划 / yo / yolov5

642

下载

yolov5 - Ultralytics YOLOv8的前身，是一个用于目标检测、图像分割和图像分类任务的先进模型。

最近提交(Master分支：3 个月前 )

6981c274 Refactor code for speed and clarity Co-authored-by: UltralyticsAssistant <web@ultralytics.com> 2 个月前

f003c3df This commit resolves an issue where the save-csv command did not write the CSV header. The code now correctly saves the header in the CSV file. Signed-off-by: Ali Ghanbari <alighanbari446@gmail.com> Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com> 2 个月前