1. 목적

2. 가정

3. 데이터 세트

3.1 데이터 세트 개요

3.2 데이터 세트의 디렉토리 구성

./data/
├── caps #データセット名
│     ├──train
│     │   ├──0.normal
│     │   │    └──正常画像1
│     │   │    └──正常画像2
│     │   │    └──正常画像3
│     │   │    ...
│     ├──test
│     │   ├──0.normal
│     │   │    └──正常画像1
│     │   │    └──正常画像2
│     │   │    └──正常画像3
│     │   │    ...
│     │   ├──1.abnormal
│     │   │    └──異常画像1
│     │   │    └──異常画像2
│     │   │    └──異常画像3
│     │   │    ...

4. 학습

4.1 Visdom 시작

python -m visdom.server

4.2 학습

 python train.py --model skipganomaly --dataset caps --abnormal_class open --display --isize 128 --device cpu --gpu_ids -1 --name capsIsOpen0123 --niter 15

5 학습 결과 확인

# LIBRARIES

from options import Options
from lib.data.dataloader import load_data
from lib.models import load_model

def main():
    opt = Options().parse()
    data = load_data(opt)
    model = load_model(opt, data)
    model.test(plot_hist=True)

if __name__ == '__main__':
    main()