train.py 训练过程中 IndexError: index 1 is out of bounds for dimension 4 with size 1

报错:IndexError: index 1 is out of bounds for dimension 4 with size 1

训练自己的模型过程中,出现以上错误
原因:数组越界

一般是类别文件没有修改

voc_classes.txt中添加子的类别

voc_classes.txt
aeroplane
bicycle
bird
boat
bottle
bus
car
cat
chair
cow
diningtable
dog
horse
motorbike
person
pottedplant
sheep
sofa
train
tvmonitor

版权声明:本文为CSDN博主「Bean_zx」的原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接及本声明。
原文链接:https://blog.csdn.net/qq_25038325/article/details/119905880

Bean_zx

我还没有学会写个人说明!

暂无评论

发表评论

相关推荐

yolov5训练数据集划分

yolov5训练数据集划分 按照默认8:1:1划分训练集,测试集,验证集。 txt文件出现在imageset文件夹。 import os import randomtrainval_pe

Pytorch—万字入门SSD物体检测

前言 由于初入物体检测领域,我在学习SSD模型的时候遇到了很多的困难。一部分困难在于相关概念不清楚,专业词汇不知其意,相关文章不知所云;另一部分困难在于网上大部分文章要么只是简要介绍了SS