yolov3-polygon icon indicating copy to clipboard operation
yolov3-polygon copied to clipboard

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

Open ccl-1 opened this issue 4 years ago • 4 comments

Traceback (most recent call last): File "E:/dl_pkg/yolov3-polygon-master/train.py", line 182, in main(opt) File "E:/dl_pkg/yolov3-polygon-master/train.py", line 117, in main loss = model(imgs.to(device), targets, requestPrecision=True) File "E:\Anaconda3\envs\torch1.3\lib\site-packages\torch\nn\modules\module.py", line 547, in call result = self.forward(*input, **kwargs) File "E:\dl_pkg\yolov3-polygon-master\models.py", line 237, in forward x, *losses = module[0](x, targets, requestPrecision) File "E:\Anaconda3\envs\torch1.3\lib\site-packages\torch\nn\modules\module.py", line 547, in call result = self.forward(*input, **kwargs) File "E:\dl_pkg\yolov3-polygon-master\models.py", line 151, in forward requestPrecision) File "E:\dl_pkg\yolov3-polygon-master\utils\utils.py", line 240, in build_targets tcls[b, a, gp_y_center, gp_x_center, tc] = 1 IndexError: index 1 is out of bounds for dimension 3 with size 1

Process finished with exit code 1 请问这个是咋回事?,类别是1,数据集标注格式为cls,x1,y1,x2,y2,x3,y3,x4,y4,theta,x,y,w,h

ccl-1 avatar Jan 08 '21 08:01 ccl-1

数据集格式,x1,y1,x2,y2,x3,y3,x4,y4,x,y,w,h 都是原始坐标,没有进行归一化

ccl-1 avatar Jan 08 '21 08:01 ccl-1

我知道错在那里了,数据类别写错了,数据集里面的数据我把它映射到0,1,2这种形式了,换成原始字符串就能训练了。

ccl-1 avatar Jan 08 '21 12:01 ccl-1

我遇到了跟你一样的问题,但是我按你说的做了还是报错

abaoji avatar Apr 27 '21 02:04 abaoji

@abaoji @ming71 @ccl-1 请问解决了吗,我把类别换成原始字符串还是报错。我的标签格式是class_name,x1,y1,x2,y2,x3,y3,x4,y4,是不是还需要加上theta,lx,ly,w,h,theta是相对于什么的角度,lxly是做左上角的坐标吗?

w-Bro avatar Jun 18 '21 09:06 w-Bro