search

gudovskiy / cflow-ad

Official PyTorch code for WACV 2022 paper "CFLOW-AD: Real-Time Unsupervised Anomaly Detection with Localization via Conditional Normalizing Flows"
https://arxiv.org/abs/2107.12571
BSD 3-Clause "New" or "Revised" License
237 stars 59 forks source link

heat map visualization #7

Closed OzanErcan6 closed 2 years ago

OzanErcan6 commented 3 years ago

First of all, thank you for presenting your work as open source. When I tested the bottle class with the

python3 main.py --gpu 0 --pro -enc mobilenet_v3_large --dataset mvtec --action-type norm-test -inp INPUT --class-name CLASS --checkpoint PATH/FILE.PT

command, I got results. DET_AUROC: last: 100.00 max: 100.00 epoch_max: 0 SEG_AUROC: last: 98.93 max: 98.93 epoch_max: 0 SEG_AUPRO: last: 96.48 max: 96.48 epoch_max: 0

The question is how can i display heatmap on images.

gudovskiy commented 3 years ago

@OzanErcan6 check this code

gudovskiy commented 2 years ago

refered to the code line

Dyy-1214 commented 2 years ago

First of all, thank you for presenting your work as open source. When I tested the bottle class with the

python3 main.py --gpu 0 --pro -enc mobilenet_v3_large --dataset mvtec --action-type norm-test -inp INPUT --class-name CLASS --checkpoint PATH/FILE.PT

command, I got results. DET_AUROC: last: 100.00 max: 100.00 epoch_max: 0 SEG_AUROC: last: 98.93 max: 98.93 epoch_max: 0 SEG_AUPRO: last: 96.48 max: 96.48 epoch_max: 0

The question is how can i display heatmap on images.

Hi, have your heatmap on images come out yet? How to debug code?

Dyy-1214 commented 2 years ago

refered to the code line

Hi, specifically how to do it? I checked the code but I don't quite understand.

gudovskiy commented 2 years ago

@Dyy-1214 you need to add --viz argument to the command above