DRAEM icon indicating copy to clipboard operation
DRAEM copied to clipboard

Inference Code

Open hamzakhalil798 opened this issue 1 year ago • 1 comments

Hey! I've added the Inference part to the code. Changes : Added a new class in dataloader.py for visualization. It loads two images instead of All images in the dataset. Added visualize_DRAEM.py, it's basically the same as test_DRAEM.py with minor changes to display Original Images and Resulting Anomaly Heatmaps.

hamzakhalil798 avatar Jan 02 '23 08:01 hamzakhalil798