284-catching-both-gray-and-black-swans-open-set-supervised-anomaly-detection-
Science Score: 10.0%
This score indicates how likely this project is to be science-related based on various indicators:
-
○CITATION.cff file
-
○codemeta.json file
-
○.zenodo.json file
-
○DOI references
-
✓Academic publication links
Links to: arxiv.org -
○Academic email domains
-
○Institutional organization owner
-
○JOSS paper metadata
-
○Scientific vocabulary similarity
Low similarity (9.1%) to scientific vocabulary
Repository
Basic Info
- Host: GitHub
- Owner: SZU-AdvTech-2023
- License: agpl-3.0
- Language: Python
- Default Branch: main
- Size: 230 MB
Statistics
- Stars: 0
- Watchers: 0
- Forks: 0
- Open Issues: 0
- Releases: 0
Metadata Files
README.md
Catching Both Gray and Black Swans: Open-set Supervised Anomaly Detection (CVPR2022)
By Choubo Ding, Guansong Pang, Chunhua Shen
Official PyTorch implementation of "Catching Both Gray and Black Swans: Open-set Supervised Anomaly Detection".
Prerequisites
This code is written in Python 3.7 and requires the packages listed in requirements.txt. Install with pip install -r
requirements.txt preferably in a virtualenv.
Run
Step 1. Setup the Anomaly Detection Dataset
Download the Anomaly Detection Dataset and convert it to MVTec AD format. (For datasets we used in the paper, we provided the convert script.)
The dataset folder structure should look like:
DATA_PATH/
subset_1/
train/
good/
test/
good/
defect_class_1/
defect_class_2/
defect_class_3/
...
...
Step 2. Running DRA
bash
python train.py --dataset_root=./data/mvtec_anomaly_detection \
--classname=carpet \
--experiment_dir=./experiment-10-30-10-90
- dataset_root denotes the path of the dataset.
- classname denotes the subset name of the dataset.
- experiment_dir denotes the path to store the experiment setting and model weight.
- outlier_root (optional) given the path of the outlier dataset to disable pseudo augmentation and enable external data for pseudo head.
- know_class (optional) specify the anomaly class in the training set to experiment within the hard setting.
Citation
bibtex
@inproceedings{ding2022catching,
title={Catching Both Gray and Black Swans: Open-set Supervised Anomaly Detection},
author={Ding, Choubo and Pang, Guansong and Shen, Chunhua},
booktitle={Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition},
year={2022}
}
Owner
- Name: SZU-AdvTech-2023
- Login: SZU-AdvTech-2023
- Kind: organization
- Repositories: 1
- Profile: https://github.com/SZU-AdvTech-2023