ILSVR2012 测试集 标签 与 mmcls imagenet.py 的不匹配,不对齐对齐问题

代码脚本结果若需要 都放在我的github:GitHub - ChenyuLy/ILSVR2012-val-label-fixContribute to ChenyuLy/ILSVR2012-val-label-fix development by creating an account on GitHub.https://github.com/ChenyuLy/ILSVR2012-val-label-fix

问题背景 :
使用mmcv mmcls 在imagenet ILSVR2012 val 数据集 做测试时
测试集  gt标签 与 mmcv  mmcls 中的 gt标签不匹配的问题

mmcls 模型中输出的是 老版的标签 只有 1k个类别
但是从官网下的数据集中 有多哒 1k+ 个类 且 类别顺序与 mmcls 数据库中的类别标签顺序对不上

方法:


使用mmcls 类中的默认CLASSES 映射到开发者工具中 meta.mat 中的 编号


首先把两张 表复制下来 其中 mmcls classes 已经是列表形式 需要把从meta.mat 复制下来的列表转化为里列表的结构方便之后的脚本处理

两文件分别是 ILSVRC2012_ID.txt  database_id.py 
对ILSVRC2012_ID.txt 用脚本格式化为 列表的形式 (这里文件不小心没留下来,需要自己格式化一下)得到 ILSVR_list

使用database_id.py完成 把 原mmcls 类中的默认CLASSES 的id 映射到 meta.mat中的编号

接着使用脚本map_gt_tobase.py 把 开发者工具中的标签ILSVRC2012_validation_ground_truth.txt 转化为  模型标签即可 得到最终正确的标签Final result.txt

同时修改mmcls中的imagenet 数据库类 如文件imagenet.py

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值