错误原因:
PASCAL5i数据集由PASCAL VOC数据集产生,将20个类分为4组,编号由0到3,每组含有5个类,官方代码在https://github.com/DeepTrial/pascal-5(与VOCtrainval_11-May-2012文件中代码相同)
官方代码生成的数据集只有第0组和第1组,并且第1组只包含一个类
错误位于config.py文件13、14行代码,以及splitDataSet.py文件23、24行代码,如下:
config.class_list和config.color_map分别用来存储20个类的名称和对应颜色,在生成每一组数据时,程序通过截取这两个数组来获得每一组的5个类。但官方代码中对数组的截取出现错误,截取的末尾漏乘5
解决办法:
将错误位置的代码进行如下修改:
config.pascal5_classlist = config.class_list[config.current_index * 5:config.current_index * 5 + 5]
config.pascal5_colormap = config.color_map[config.current_index * 5:config.current_index * 5 + 5]
即在数组截取末尾乘上5