mmdetection训练出现：IndexError:listindexoutofrange错误

我的环境&＃xff1a;

• Ubuntu18.04
• TorchVision: 0.6.0
• OpenCV: 4.2.0
• MMCV: 0.5.5
• MMDetection: 2.0.0&＃43;d9c8f14
• MMDetection Compiler: GCC 7.5
• MMDetection CUDA Compiler: 10.2

训练的模型是&＃xff1a;Retinanet

1 问题分析

在使用mmdetection2.0框架训练目标检测模型时候&＃xff0c;出现IndexError: list index out of range错误&＃xff0c;具体内容如下图&＃xff0c;首先从这个错误的类型我们可以看出是索引超过了列表的长度&＃xff0c;导致IndexError错误。

1.1 尝试解决错误&＃xff1a;第一次

根据这个错误&＃xff0c;很容易想到是否是类别写的不正确&＃xff0c;于是去检查如下三个文件&＃xff1a;

• mmdetection/mmdet/datasets/voc.py&＃xff1a;中的CLASSED变量对应的类别是否正确
• mmdetection/mmdet/core/evaluation/class_names.py&＃xff1a;voc_classes()函数返回的类别是否正确
• mmdetection/configs/_base_/models/retinanet_r50_fpn.py&＃xff1a;中num_classes对应的类别数是否正确

结果反复检查了好几遍&＃xff0c;还是没有解决这个问题&＃xff0c;当然如果你遇到这个问题&＃xff0c;很可能就是因为类别写的不对&＃xff0c;你一定要仔细、仔细、仔细检查。

1.2 尝试解决错误&＃xff1a;第二次

自己解决不了&＃xff0c;于是就去逛github mmdetection的官网&＃xff0c;然后乖乖查看issues&＃xff0c;issues中好多说是类别的问题&＃xff0c;还有一个人说是学习率太大导致的&＃xff0c;我一脸迷惑&＃xff0c;信以为真&＃xff0c;就尝试了一下&＃xff08;参考&＃xff09;

ONE YEARS LATER

一顿尝试&＃xff0c;最终还是不行&＃xff0c;错误重现&＃xff01;&＃xff01;&＃xff01;

2 我的问题解决方式

我最开始编译的mmdetection环境路径为&＃xff1a;

/home/shl/shl/mmdetection

但是我后面训练自己的另外一个数据集的时候&＃xff0c;是直接把这个编译好的mmdetection拷贝一份出来&＃xff0c;会出一定的问题&＃xff0c;比如&＃xff1a;

cp -r mmdetection mmdetection_clothes

然后在/home/shl/shl/mmdetection_clothes中修改类别&＃xff0c;但是最终还是会去&＃xff1a;/home/shl/shl/mmdetection/mmdet下面读取类别信息&＃xff0c;用pip list 可以查看到mmdetection的路径

(mmdetection) shl&＃64;zfcv:~/shl/mmdetection$ pip list
Package Version Location
---------------------- ------------------- -------------------------
absl-py 0.9.0
addict 2.2.1
cachetools 4.1.0
certifi 2020.4.5.1
chardet 3.0.4
click 7.1.2
cycler 0.10.0
Cython 0.29.19
Flask 1.1.2
future 0.18.2
google-auth 1.17.2
google-auth-oauthlib 0.4.1
grpcio 1.29.0
idna 2.9
importlib-metadata 1.6.1
itsdangerous 1.1.0
Jinja2 2.11.2
kiwisolver 1.2.0
Markdown 3.2.2
MarkupSafe 1.1.1
matplotlib 3.2.1
mmcv 0.5.5
mmdet 2.0.0&＃43;d9c8f14 /home/shl/shl/mmdetection # 这里可以看到&＃xff0c;mmdetection_clothes训练时还是在这里调用mmdet下的类别相关信息
numpy 1.18.4
oauthlib 3.1.0
opencv-python 4.2.0.34
pandas 1.0.4
Pillow 7.1.2
pip 20.0.2
protobuf 3.12.2
pyasn1 0.4.8
pyasn1-modules 0.2.8
pycocotools 2.0
pyparsing 2.4.7
python-dateutil 2.8.1
pytz 2020.1
PyYAML 5.3.1
requests 2.23.0
requests-oauthlib 1.3.0
rsa 4.6
scipy 1.4.1
seaborn 0.10.1
setuptools 46.4.0.post20200518
six 1.15.0
tensorboard 2.2.2
tensorboard-plugin-wit 1.6.0.post3
terminaltables 3.1.0
torch 1.5.0
torchvision 0.6.0
tqdm 4.46.1
urllib3 1.25.9
Werkzeug 1.0.1
wheel 0.34.2
yapf 0.30.0
zipp 3.1.0
(mmdetection) shl&＃64;zfcv:~/shl/mmdetection$


所以最后的解决方式&＃xff0c;在/home/shl/shl/mmdtection这个原始的路径下修改类别&＃xff0c;然后训练&＃xff0c;不要拷贝出来一份&＃xff01;我就是想训练多个数据集才这样干的&＃xff0c;至于后面有没有什么解决方式&＃xff0c;我再尝试&＃xff01;