Spaces:
Running
on
Zero
Running
on
Zero
updating readme
Browse files
README.md
CHANGED
@@ -9,4 +9,15 @@ app_file: app.py
|
|
9 |
pinned: false
|
10 |
---
|
11 |
|
12 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
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.
|