ahsanMah commited on
Commit
deb6231
·
1 Parent(s): f1e86bd

updating readme

Browse files
Files changed (1) hide show
  1. README.md +12 -1
README.md CHANGED
@@ -9,4 +9,15 @@ app_file: app.py
9
  pinned: false
10
  ---
11
 
12
- Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
 
 
 
 
 
 
 
 
 
 
 
 
9
  pinned: false
10
  ---
11
 
12
+ # Anomaly Localization with Score-Based Diffusion Models
13
+
14
+ Detect anomalies in natural images!
15
+
16
+ ### Background
17
+ This project aims to add anomaly detection capabilites to score-based diffusion models. The main idea is to learn the distribution of the typical score vectors for each patch position. One can train a postion-condiditoned normalizing flow model to estimate the likelihood of the outputs of the score vector. The upshot is that you can visualize the per-patch likelihood and inspect the resulting heatmap of anomalies! This idea comes from work done as part of my doctoral dissertation, with the underlying methodology described in (this paper)[https://arxiv.org/abs/2010.13132].
18
+
19
+
20
+ The code builds upon the excellent (EDM2 repository)[https://github.com/NVlabs/edm2/] by NVLabs. Please refer to the original work form more information on the diffusion models.
21
+
22
+ ### Caveats
23
+ As the underlying models are trained on Imagenet-1k, you may have better success when the subject belongs to one of the 1000 classes it was trianed on.