본문 바로가기

728x90
반응형

tensorflow

(2)
[CNN] LeNet-5 모델 구현하기 LeNet-5는 손글씨, 우편 인식 등 이미지 분류에 쓰이던 DNN(Deep Neural Network)이다. LeNet-5에 대한 자세한 정보는 [1]에서 찾아 볼 수 있지만, 간단하게 설명한 후에 Tensorflow로 어떻게 구현해야 하는지 코드 리뷰를 하겠다. LeNet-5는 입력층을 제외하고 3개의 Convolutional Layer(Cx), 2개의 Sub-Sampling Layer(Sx), 그리고 2개의 Full-Connected Layer(Fx)로 총 7개의 층으로 구성되어 있다. 그래서 [그림 1]에서 C1 - S2 - C3 - S4 - C5 - F6 - OUTPUT(F7)와 같이 층이 쌓이는 순서대로 표현한 걸 볼 수 있다. Sub-Sampling Layer는 Pooling Layer라고도..
[Keras] Keras 내장 모델 불러오기 또는 모델 구조만 불러오기 Dacon 대회에 참여 할 때 높은 정확도를 추출하고자 증명된 모델을 사용하고자 했다. 하지만 대회 규정에는 이미 학습된 모델을 사용하는 것은 위반이기 때문에 모델 구조만 가져오는 법을 찾다가 정리 또는 기록 겸 글을 작성한다. 모델 구조만 불러오는 방법과 학습된 모델를 불러오는게 별 차이가 없어서 같이 기록하려고 한다. Keras에는 기본적으로 ImageNet 데이터셋에 대해 학습된 모델을 제공해주고 있다. 사용 가능한 모델은 다음과 같다. Xception VGG16, VGG19 ResNet, ResNetV2, ResNeXt InceptionV3 MobileNet DenseNet NASNet 이제 위 모델을 불러올 때 1)학습된 모델 또는 2)모델 구조만 불러오는 방법을 알아보자. 1) import i..

728x90
반응형