-
Notifications
You must be signed in to change notification settings - Fork 13
Open
Description
高老师您好,我以The Changes You Want To See(CYWS)的代码框架为基础尝试复现MetaUAS的结果。训练数据集只使用InpatinedCocoDataset,数据增强方式大部分保持CYWS的数据增强模式,同时遵循DRAEM论文中的方法,增加了使用Perlin噪声合成的异常,以实现local region changes。
模型方面完全遵循MetaUAS的配置,使用了代码提供的模型。学习率设置为1e-4,Weight Decay 为,BatchSize设置为128,训练30epochs。然后直接使用这个模型在MVTecAD数据集上进行测试。模型测试代码与MetaUAS的配置完全相同,使用oneprompt_seed1.json配置。
训练结果为:
| Name | I-AUROC | I-AP | I-F1max | P-AUROC | P-AP | P-F1max | P-AUPRO |
|---|---|---|---|---|---|---|---|
| bottle | 0.974 | 0.992 | 0.961 | 0.901 | 0.537 | 0.486 | 0.630 |
| cable | 0.650 | 0.790 | 0.763 | 0.927 | 0.338 | 0.374 | 0.710 |
| capsule | 0.649 | 0.885 | 0.905 | 0.804 | 0.088 | 0.135 | 0.366 |
| carpet | 0.878 | 0.968 | 0.909 | 0.914 | 0.514 | 0.512 | 0.555 |
| grid | 0.815 | 0.932 | 0.862 | 0.785 | 0.160 | 0.268 | 0.419 |
| hazelnut | 0.951 | 0.971 | 0.923 | 0.982 | 0.754 | 0.705 | 0.909 |
| leather | 0.999 | 1.000 | 0.995 | 0.979 | 0.406 | 0.419 | 0.930 |
| metal_nut | 0.708 | 0.907 | 0.912 | 0.540 | 0.224 | 0.257 | 0.494 |
| pill | 0.601 | 0.905 | 0.916 | 0.606 | 0.142 | 0.251 | 0.459 |
| screw | 0.611 | 0.827 | 0.858 | 0.779 | 0.025 | 0.059 | 0.487 |
| tile | 0.846 | 0.941 | 0.865 | 0.820 | 0.507 | 0.480 | 0.620 |
| toothbrush | 0.744 | 0.904 | 0.845 | 0.820 | 0.296 | 0.382 | 0.493 |
| transistor | 0.628 | 0.579 | 0.608 | 0.798 | 0.207 | 0.279 | 0.527 |
| wood | 0.988 | 0.996 | 0.968 | 0.894 | 0.541 | 0.543 | 0.622 |
| zipper | 0.686 | 0.906 | 0.881 | 0.692 | 0.090 | 0.119 | 0.277 |
| Avg | 0.782 | 0.900 | 0.878 | 0.816 | 0.322 | 0.351 | 0.567 |
与论文的结果相差比较大。
是不是我在复现的过程中忽略了某些重要的细节?例如需要进行梯度裁剪,或者需要在训练数据集中加入一部分MVTec-AD的训练数据集一起训练。
Reactions are currently unavailable
Metadata
Metadata
Assignees
Labels
No labels