Skip to content

異常検知手法ALOCCの実装 (Chainer, PyTorch)

License

Notifications You must be signed in to change notification settings

kzkadc/alocc_mnist

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

16 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

ディープラーニングによる異常検知手法ALOCC

Sabokrou, et al. "Adversarially Learned One-Class Classifier for Novelty Detection", The IEEE Conference on Computer Vision and Pattern Recognition (CVPR), 2018, pp. 3379-3388

https://arxiv.org/abs/1802.09088

準備 (Chainer)

ChainerとOpenCVを使います。
インストール:

$ sudo pip install chainer opencv-python

準備 (PyTorch)

PyTorch, PyTorch-Ignite, OpenCV, Matplotlib, scikit-learnを使います。

インストール:
PyTorch: 公式HPを参照

$ sudo pip install pytorch-ignite opencv-python matplotlib sklearn

実行

設定ファイルsetting.jsonを編集することでハイパーパラメータを設定できます。

$ python train.py 設定ファイル 出力フォルダ [-g GPUID]

例:

$ python train.py setting.json result

About

異常検知手法ALOCCの実装 (Chainer, PyTorch)

Topics

Resources

License

Stars

Watchers

Forks

Languages