你们的开源代码质量很高,值得学习!我想请问,samples文件下的json文件的生成脚本能否上传一下呢?
EWAN9709 opened this issue · 10 comments
EWAN9709 commented
你们的开源代码质量很高,值得学习!我想请问,samples文件下的json文件的生成脚本能否上传一下呢?
cnulab commented
感谢您对我们工作的支持!
以下的脚本演示了如何为MVTec数据集中的bottle类生成对应的json文件,其他数据集类似,以供您参考。
create_sample_json.zip
EWAN9709 commented
谢谢大佬!
EWAN9709 commented
cnulab commented
你好,这是由于你没有为你的自定义数据集设置对应的前景分割方案,一个简单的方法是去除前景分割:
- 注释掉 datasets/realnet_dataset.py 中的239行 target_foreground_mask = self.generate_target_foreground_mask(img,dataset, subclass)
- 修改246行,直接让 mask = perlin_noise_mask
去掉前景分割可能会带来一定的性能损失
EWAN9709 commented
cnulab commented
你好,由于自定义数据集没有生成对应的异常图像,推荐你使用DTD数据集进行异常合成:
- 下载DTD数据集并按照data/README.md中的说明将数据集放置在正确的位置。
- 修改配置文件,改为使用DTD数据集。例如,注释掉 experiments/MVTec-AD/realnet.yaml 中的23~25行,使用28~30行。
EWAN9709 commented
谢谢大佬的指点,在私有数据集指标也很高,再次感谢你们贡献了如此优秀的工作,向你们学习!
EWAN9709 commented
cnulab commented
改善的建议不好说,需要具体问题具体分析。你的缺陷区域占输入图像的多大面积,训练样本中是否包含类似的缺陷区域可能误导网络的训练。该算法依赖于合成异常进行网络训练,如果你有大量的缺陷样本可以直接使用有监督的训练,这是提升性能的最快方式。此外,如果你的数据集中都是小面积的缺陷区域,可以尝试调整一下网络结构,修改experiments/MVTec-AD/realnet.yaml中的structure.
EWAN9709 commented
好的,谢谢,我去试试