search

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

add gradcam #13

Closed lg93lggt closed 2 years ago

lg93lggt commented 2 years ago

add gradcam and have a little change on model.py. Usage is in ./test/test_grad_cam.py