open-mmlab/mmdetection

推理时的小bug

jamesbondzhou opened this issue · 0 comments

我用自己的数据集训练了一个模型,这个数据集只有一个类别smoke,在推理的时候,模型输出可视化结果那里,目标的类别并不是smoke,而是s,见下图:
企业微信截图_17326063503455

按照官方的说明准备的配置文件

修改数据集相关配置

data_root = 'data/smoke/'
metainfo = {
'classes': ('smoke', ),
'palette': [
(220, 20, 60),
]
}
想问下究竟是哪里出了问题呢,而且我之前训练多类别的数据时一点问题都没有,我也试了分别用两种算法训练的模型来推理,结果都是这样,很奇怪!项目我也更新了,环境也都更新了