anomaly-detection-using-autoencoders

KO | EN

gitlite — search

#python #go #machine-learning #deep-learning #docker #macos #java #nodejs #windows #awesome #cli #kubernetes #security #mcp

anomaly-detection-using-autoencoders

(★ 64)

This is the implementation of Semi-supervised Anomaly Detection using AutoEncoders

anomaly-detection-using-autoencoders 최신버젼 다운로드

최종 버전 다운로드 (.zip)

// repository documentation

Was this content helpful?

★ 0 (0 ratings)

This is the implementation of [Semi-supervised Anomaly Detection using AutoEncoders](https://arxiv.org/abs/2001.03674 "Semi-supervised Anomaly Detection using AutoEncoders")

The hypothesis of the paper is that an AutoEncoder trained on just the defect free or normal samples will fail to reconstruct the images that have defects in it since those were not seen during training. This will lead to large residuals which can be then be used to segment the anomaly.

For this repository the dataset needs to be organised as:
```
TrainDir
----NonDefect
    |----Img1
	|    ...
    |----ImgM
ValDir
----NonDefect
    |----Img1
	|	...
    |----ImgN
```

Usage:
```bash
python main.py --train_dir /path/to/train/dir/ --val_dir /path/to/val/dir
```

```
Requirements:
PyTorch >= 1.4
Ignite 0.3.0
Tensorboard 2.1.0
```

An example segmentation output of AutoEncoder trained on the [DAGM dataset](https://hci.iwr.uni-heidelberg.de/content/weakly-supervised-learning-industrial-optical-inspection "DAGM dataset") is shown below. The model was trained on the Class8 and trained for 25 epochs. The best model was kept based on the lowest mean squared error value.

![Sample segmentation output](https://raw.githubusercontent.com/msminhas93/anomaly-detection-using-autoencoders/master/image/sample_detection.png)

anomaly-detection-using-autoencoders

anomaly-detection-using-autoencoders 최신버젼 다운로드

오늘의 추천 프로젝트

Was this content helpful?

Recent Feedback

anomaly-detection-using-autoencoders

anomaly-detection-using-autoencoders 최신버젼 다운로드

오늘의 추천 프로젝트

Was this content helpful?

Recent Feedback

Download README