CycleGAN 모델 파라미터 정리

어느 모델을 돌릴 때나 본인의 환경에 맞게 파라미터를 사용하는데 cyclegan도 마찬가지이다.

보통은 유명한 모델일수록 model에 관련된 설명이 github에 잘 나와있기 때문에 cyclegan도 dataset 경로, 모델 돌릴 때의 tips, 파라미터 설명 등이 github에 아주 잘 올라와 있지만,

cyclegan을 train/test 할 때 많이 사용하는 파라미터를 기록용으로 옮겨 적어두고자 한다. 

 

각 옵션들은 base_options.py, train_options.py, test_options.py 에 나눠져 있다. 

---

base options

--dataroot: 데이터셋 이미지 경로 지정. trainA, trainB, valA, valB 등의 하위 폴더가 있어야 함

• default 값: 없음 (required=True)

--name: 실험의 이름 지정. 샘플과 모델을 저장할 위치를 결정함

• default 값: 'experiment_name'

--gpu_ids: 사용할 GPU ID 지정. 예: 0, 0,1,2, 0,2. CPU 사용 시 -1을 입력

• default 값: '0'

--checkpoints_dir: 모델이 저장될 디렉토리 지정

• default 값: './checkpoints'

--norm: 정규화 방법 선택. instance, batch, none 중 선택 가능

• default 값: 'instance'

--batch_size: 입력 배치 크기 지정

• default 값: 1

--load_size: 이미지를 이 크기로 스케일링

• default 값: 286

--crop_size: 스케일링 후 이 크기로 이미지를 자름

• default 값: 256

--preprocess: 이미지 로드 시 스케일링 및 자르기 방법 선택. resize_and_crop, crop, scale_width, scale_width_and_crop, none 중 선택 가능

• default 값: 'resize_and_crop'

--epoch: 로드할 epoch 지정. 'latest'로 설정하면 최신에 저장된 모델을 사용

• default 값: 'latest'

--use_wandb: 지정하면 wandb 로깅을 초기화

• default 값: False (action='store_true')

--wandb_project_name: wandb 프로젝트 이름 지정

• default 값: 'CycleGAN-and-pix2pix'

train options

--lr: adam 옵티마이저의 초기 학습률 지정

• default 값: 0.0002

--lr_policy: 학습률 정책 선택. linear, step, plateau, cosine 중 선택 가능

• default 값: 'linear'

--lr_decay_iters: 이 반복 횟수마다 감마를 곱함

• default 값: 50

test options

--results_dir: 결과를 저장할 디렉토리 지정

• default 값: './results/'

 

 

Reference

https://github.com/junyanz/pytorch-CycleGAN-and-pix2pix/blob/master/options