xcyao00/BGAD

custom dataset异常区域定位问题

Yibo-0820 opened this issue · 1 comments

非常感谢你们的工作!
由于我刚接触异常检测相关知识,有一些问题不理解,想求教一下
我想在我自己的数据集上尝试异常区域的定位,我想问下,想要训练出好的效果,数据集缺陷的groundtruth的mask是否是必须用于监督(因为我自己数据集并没有缺陷部位mask的图片,只是单纯的正常图片和异常图片)
因为我看缺陷部位mask的GT是用于在训练过程中计算bg_sppc loss?我只是为了得到模糊的,而不是精细的异常部位定位输出的话,是否可以不需要用它?

我们的方法是基于特征图上的特征的,如果你想训练时使用异常,是需要异常的mask的;但是,你如果在训练时不需要使用异常,只是基于正常训练,是可以不需要异常mask的。这样你就只用正常样本训练第一阶段就行了。