PASCAL-5i数据集无法产生完整数据集的解决办法

错误原因:

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

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值