- 파이토치 Image-To-Image Translation With Conditional Adversarial Networks
- 학습
- python -m visdom.server 실행
- python Main/run_train.py 실행
- 테스트
- python Main/run_test.py 실행
- 학습 및 테스트
- 모델 파일 저장 및 불러오기
- 학습 진행 과정 그래프로 시각화
- 학습 결과물 이미지로 저장
- 데이터 변환 방향 선택하여 학습 가능 (a2b 또는 b2a)
- Main/run_train.py 및 Main/run_test.py 에서 디바이스, 모델, optimizer, dataloader, 손실 함수, metric 등 모두 선언 및 실행
- DeepLearning/layer.py 에 모델을 구성하기 위한 nn.Module 들을 미리 만들어 놓고 DeepLearning/model.py 에서 모델을 구성하는데 사용
- Training progress
