Source code

[sumeetssaurav]

Hii,

Thank you for the interesting work on the industrial anomaly detection. I am currently working on the similar problem and would like to try for proposed framework for comparison. Therefore, I request you to kindly provide the source code to train and test the models on the datasets and validate the results.

Thanking you
Sumeet

[andreluizbvs]

Unfortunately the code for AttentDifferNet is not publicly available. However, to reproduce the method it is simple:

1. Clone the DifferNet repository
2. Go here and replace this call to the alexnet backbone with its actual architecture, which can be found here.
3. Add the attention layers (SENet or CBAM or any other) in the positions described in the paper (inside self.features of the alexnet architecture).

I hope this can help you.