KO | EN

GitLite

Minimalist Open Source Repository Search Engine

gitlite — search

#developer-tools #typescript #dsh #dsh-plugin #ai-agent #codex #ai-agents #deepseek-harness #claude-code #cli #react #mcp

PromptAD

★ 212 Open GitHub ↗

Official implementation of CVPR 2024 PromptAD: Learning Prompts with Only Normal Samples for Few-Shot Anomaly Detection

// repository documentation

Was this content helpful?

★ 0 (0 ratings)

PromptAD Few-Shot Anomaly Detection
=================================
Official implementation of [PromptAD: Learning Prompts with only Normal Samples for Few-Shot Anomaly Detection](http://arxiv.org/abs/2404.05231) (CVPR2024)

![RUNOOB 图标](https://github.com/FuNz-0/PromptAD/blob/master/PromptAD.jpg)

## Install
```
conda create -n prompt_ad python==3.10
conda activate prompt_ad
bash install.sh
```
## Data
Download the dataset from [MvTec](https://www.mvtec.com/company/research/datasets/mvtec-ad).

Download the dataset from [VisA](https://github.com/amazon-science/spot-diff?tab=readme-ov-file#data-download).

#### VisA preprocessing
Modify the source and target paths for the VisA dataset in `./dataset/prepare_visa_public.py`
```
python ./dataset/prepare_visa_public.py
```
Modify the source paths for MvTec and VisA in `./dataset/mvtec.py` and `./dataset/visa.py`
## Run
```
python run_cls.py  # image-level
python run_seg.py  # pixel-level
```

## Citation
Please cite the following paper if this work helps your project:
```
@article{li2024promptad,
  title={PromptAD: Learning Prompts with only Normal Samples for Few-Shot Anomaly Detection},
  author={Li, Xiaofan and Zhang, Zhizhong and Tan, Xin and Chen, Chengwei and Qu, Yanyun and Xie, Yuan and Ma, Lizhuang},
  journal={arXiv preprint arXiv:2404.05231},
  year={2024}
}
```

## Acknowledge

We thank the great works [WinCLIP](https://github.com/caoyunkang/WinClip.git) and [CoOp](https://github.com/KaiyangZhou/CoOp.git) for assisting with our work.

PromptAD

Explore Similar Repositories

Was this content helpful?

Recent Feedback

Download README