sjtuplayer / anomalydiffusion

[AAAI 2024] AnomalyDiffusion: Few-Shot Anomaly Image Generation with Diffusion Model
MIT License
138 stars 19 forks source link

The training and testing of segmentation model #11

Open DaiZhewei opened 6 months ago

DaiZhewei commented 6 months ago

Hi author, I would like to ask two questions:

  1. whether the method is divided into 15 classes to train the classification model and segmentation model respectively
  2. how to divide the training set and test set? Looking forward to your answer, thanks!
sjtuplayer commented 5 months ago
  1. Yes, we train one classification model and segmentation model for each sample class (e.g., bottle, capsule) separately.
  2. The training set is the generated data while the test data is the last 2/3 data in the test set of MvTec.
sjtuplayer commented 5 months ago

Hi author, I would like to ask two questions:

  1. whether the method is divided into 15 classes to train the classification model and segmentation model respectively
  2. how to divide the training set and test set? Looking forward to your answer, thanks!

You can refer to ''train-unet.py", "test-unet.py" and "train-classification.py" for more details

DaiZhewei commented 5 months ago

thanks, one more question, did you use all the masks in the test set when training the TI or only the first 1/3 of the masks?

engrmusawarali commented 5 months ago

@DaiZhewei Thanks.

DaiZhewei commented 5 months ago

@DaiZhewei could you send your email address we may discuss some things, hi, I e-mailed the e-mail address on your home page.

CVKim commented 3 weeks ago

@sjtuplayer

Hi author,

It seems that the file train-unet.py is missing. Could it have been removed?

sjtuplayer commented 3 weeks ago

@sjtuplayer

Hi author,

It seems that the file train-unet.py is missing. Could it have been removed?

Hi. It is train-localization.py in the current version.

CVKim commented 3 weeks ago

In the "Anomaly Diffusion" paper, is it correct to say that the performance evaluation focused more on anomaly detection, specifically localization and classification, rather than on segmentation?

Test-localization and test-classification.

image