🚧 WORK (still) IN PROGRESS 🚧

Quick Overview of how to train PNI:

• Use the train_PNI method (in patchcore.py) to train the MLP and record histograms any number of epochs
• Use the PNI_predict (in patchcore.py) method for (obviously) predicting
• Use the refinement_model.py which is the exact refinement architecture as the paper
• Dataloader used here is a custom dataloader (not MVTech)
• There is also a custom cutpaste transformation that also generates GTs