딥러닝 과목 기말 프로젝트 코드입니다. 다양한 컬러 숫자이미지를 인식할 수 있는 CNN모델 코드와 MNIST를 이용해 패턴 컬러 이미지를 만드는 data augmentation 코드로 이루어져 있습니다.
28*28 사이즈의 사진, 그림 등으로 이루어진 컬러 숫자이미지를 인식하는 CNN 모델 제작. general model을 위한 training data augmentation. 모델에 대한 분석.
- Anaconda 3
- python 3.9
- pytorch
- torchvision
패턴 이미지와 MNIST 데이터셋을 4가지 방법으로 조합하여 training data 제작. 각 이미지별 label data 기록.
(3*3 conv2d + ReLU + MaxPooling) x2, (Fully connected + ReLU) x3
모델의 디자인과 성능에 대한 정보는 analysis.pdf
에서 확인할 수 있습니다.