mjq11302010044/RRPN

训练网络时报错,难道是数据生成的不对吗?

19931991 opened this issue · 1 comments

学长,你好,我按照你在github上的步骤一步步进行,demo测试没有问题,但是在训练网络时报错,在网上也搜不到答案,请问下面的问题是数据生成的不对吗?
数据数据时用到的代码内容如下:
image
image

然后运行生成数据的代码,运行结果是这样的:
image

train_net.py网络中的第116行修改如下:
image

然后就运行 ./data/scripts/fetch_imagenet_models.sh 下载Pre-trained ImageNet models ,有没有报错,紧接着运行 ./experiment/scripts/faster_rcnn_end2end.sh 0 vgg16 rrpn ,就出现了如下错误:

image

请问学长是数据的问题吗?@
Owner
mjq11302010044

补充说明一下,我用的数据是ICDAR2015的,自己组织数据时只使用到了/lib/rotation/data_extractor.py 中的get_rroidb() 和 get_ICDAR2015_RRC_TRAIN(mode)函数,没有使用其他的函数。不知道我自组织的数据对不对?还请学长指点。非常感谢@mjq11302010044