wogur110 / PNI_anomaly_detection

Official code for the paper "PNI : Industrial Anomaly Detection using Position and Neighborhood Information"
30 stars 6 forks source link

Hi, I see that in your paper you used three types of synthetic defect images to train the pixelwise refinement network, but I didn’t find the relevant ones in the code. Can you point out which file the specific code is in? #2

Closed sevenn126 closed 1 week ago

wogur110 commented 8 months ago

Hi. main/pixelwise_refinement.py is the specific code what you mentioned.

StefanoSamele-PoliMi commented 6 months ago

Hi, thank you for sharing your exciting work. I was wondering if you could also share the code to prepare the images used to train the refinement network. My understanding is that main/pixelwise_refinement.py is just a script that does the training/evaluation, but the images are pre-prepared and stored under the location indicated by --input_data_path argument.

wogur110 commented 6 months ago

Hi, I prepared the images for training the refinement network using 1) DREAM, 2) CutPaste, and 3) manual data. For DREAM and CutPaste, I made the dataset using existing official codes, so it is not necessary to share the codes. For manual data, I can share synthetic datasets, but I'm not sure if I can share MVTec data directly. If you want, I will share the manual data only.

StefanoSamele-PoliMi commented 6 months ago

Thank you for the follow-up. I am not interested in the specific MVTec dataset but rather in the procedure you adopted for creating the manual data: can I kindly ask you if you can give more details about the types of drawings and synthetic anomalies you inserted? Do they change from class to class? Maybe just a few key examples could let us understand. Thank you in advance for your support.