xcyao00/BGAD

custom my own dataset 如何制造我自己的数据集?

Opened this issue · 2 comments

非常感谢你们的工作!
由于我刚接触异常检测相关知识,有一些问题不理解,想求教一下
在论文中你提到了elpv数据集,但是在代码中我没有看到如何处理的。我想问下我该如何处理这些数据集得到groundtruth下的那些mask图像呢?
谢谢!

elpv数据集并没有提供pixel-levl的mask标注,所以没法得到mask图像;对于elpv,我们只能进行image-level的异常分类,我们在elpv数据集上也是测试了image-level AUROC的结果。对于没有mask图像的数据集,我们将backbone最后pooled的全局特征输入归一化流模型进行学习。

我想请问一下我该如何处理elpv的数据集?对于elpv代码而言,是不是意味着我得到的normal_dataset和train_dataset是一样的吗?可以分享一下你关于处理elpv数据集部分的代码吗?
谢谢!