PyTorch-Autoencoder using MNIST

https://qiita.com/satolab/items/8efa513e7fd6cb41fdc5

Powered by satolab

Overview

PyTorchを用いた，AutoencoderによるMNISTの異常検知プログラムです．

Model

Results

• 10 epochs(input,output,difference)

Usage

• main.pyで学習．save dirにサンプルが保存されます． Learn with main.py. The sample is saved in save dir.

References

差分画像の計算と表示部分 http://cedro3.com/ai/keras-autoencoder-anomaly/