Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

I can not reproduce your results by using the uploaded checkpoints #23

Closed
sunchengyi opened this issue Feb 20, 2024 · 3 comments
Closed

Comments

@sunchengyi
Copy link

I am trying to reproduce the results by using the uploaded checkpoints on hazelnut and screw and can not get the same scores.

For hazelnut the output is as follows:
Class: hazelnut w: 5 v: 1 load_chp: 2000 feature extractor: wide_resnet101_2 w_DA: 3 DLlambda: 0.1
config.model.test_trajectoy_steps=250 , config.data.test_batch_size=16
Detecting Anomalies...
AUROC: (99.4,98.3)
PRO: 87.1

For screw, it is:
Class: screw w: 2 v: 1 load_chp: 2000 feature extractor: wide_resnet101_2 w_DA: 3 DLlambda: 0.1
config.model.test_trajectoy_steps=250 , config.data.test_batch_size=16
Detecting Anomalies...
AUROC: (96.5,99.3)
PRO: 96.5

It seems that Pixel_AUROC and PRO are similar to your results, but Image_AUROC is very different.
Is there something wrong?
I just ran the code by using the checkpoints. My Python is 3.10 and torch is 2.0.1.

@arimousa
Copy link
Owner

I'm uncertain if this could be the cause, but could you please switch the Python version to 3.8 and rerun the code?

@sunchengyi
Copy link
Author

sunchengyi commented Feb 21, 2024

Thank you for the reply. I will try it.
Actually, I have tried other AD models, too, and can not get the same scores as in papers,
particularly when I trained the models myself. However, on MVTecAD, I got the best scores from DDAD by training them myself.
Thank you very much! Great Work!

@arimousa
Copy link
Owner

Glad to hear that.
Then I will close the issue.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants