search

zqhang / AnomalyCLIP

Official implementation for AnomalyCLIP (ICLR 2024)
MIT License
285 stars 30 forks source link

作者大大和各位专家，请问如何设置自己的数据集？ #35

Open LJK12581 opened 4 months ago

LJK12581 commented 4 months ago

亲爱的作者大大，以及各位专家和大佬，请问我如果要训练自己的数据集应该如何操作？假设我有一堆图片，那我是否需要掩码呢？在原作者给出的七个工业数据集上，每个数据集的设置都多少有些不同，有的有ground_truth，有的则没有。请问如果我要设置自己的数据集，应该具体如何操作？图片的大小和分别率等是否有要求？还有，如果我设置好数据集之后，在运行测试结果的时候我暂时不需要测试模型的四个指标，仅仅需要判断给出的测试值中的数据哪些是正常，哪些是异常即可，请问我应该修改哪里的代码？又应该如何修改呢？ 下面是我设置的数据集目录格式，请问是否有误？以上所有问题，真诚的向作者，各位专家大佬谦逊提问，望解答！ image

zqhang commented 4 months ago

1、AnomalyCLIP的训练数据是需要groudtruth 2、数据集的结构其实可以是任意的，只要你在generate_dataset_json文件夹里写一个相应的脚本就好了。如果你觉得麻烦，你可以参照MVTec的数据集格式调整你的数据集，这样你就可以直接用mvtec.py生成相应的json文件来读取数据集了 3、测试样本是正常还是异常的话，你需要修改test.py倒数几行的代码。AnomalyCLIP输出是一个衡量异常程度的分数。如果你要得到异常还是正常的离散结果，要根据你的测试结果设定一个阈值或者其他的方法。

LJK12581 commented 4 months ago

1、AnomalyCLIP的训练数据是需要groudtruth 2、数据集的结构其实可以是任意的，只要你在generate_dataset_json文件夹里写一个相应的脚本就好了。如果你觉得麻烦，你可以参照MVTec的数据集格式调整你的数据集，这样你就可以直接用mvtec.py生成相应的json文件来读取数据集了 3、测试样本是正常还是异常的话，你需要修改test.py倒数几行的代码。AnomalyCLIP输出是一个衡量异常程度的分数。如果你要得到异常还是正常的离散结果，要根据你的测试结果设定一个阈值或者其他的方法。

感谢您忙里抽闲回复我的问题，我的困惑迎刃而解。如有其他问题向您请教，后续我再向您提问，再次感谢！