LilitYolyan / CutPaste

Unofficial implementation of Google "CutPaste: Self-Supervised Learning for Anomaly Detection and Localization" in PyTorch
MIT License
114 stars 25 forks source link

Bug in localization #27

Open Chichiviriche opened 2 years ago

Chichiviriche commented 2 years ago

Hello, can you help: heatmap_on_image() in the final line of code of localization.py has 3 arguments, but while defining it, there were only 2 arguments. Also visualize_heatmap() is not defined Thanks

aytek03 commented 2 years ago

Hi @Chichiviriche

You can use save_anomaly_map() instead of visualize_heatmap().

heatmap_on_image() takes two argument. One is image another is Gaussian_smoothing() from patch_core.

Chichiviriche commented 2 years ago

Hi, thanks for your answer. Yes I took save_anomaly_map(). Anyway I didn't succeed to have localization.py working properly : I got a blue image as the heat map I saw somewhere else that there are other bugs in localization.py : do you have a working version of localization.py ? Regards,

aytek03 commented 2 years ago

Hi @Chichiviriche

I got the same results. That is just blue heat map.

Chichiviriche commented 1 year ago

Hi, does someone succeed to tune localization.py to make it work ? Help would be appreciated. kind regards,

18894269590 commented 1 year ago

@Chichiviriche Hi, did you success to tune localization.py to make it work ?

18894269590 commented 1 year ago

@Chichiviriche I also value this code and have done some work based on it, can we have a deeper conversation, my email is qswang20@fudan.edu.cn

Chichiviriche commented 1 year ago

no progress on localization.py on my side. Did you make any good steps ?