< 출처 : http://loenahmik.blogspot.kr/2016/09/windows-caffe.html >
테스트 환경은 아래와 같습니다.
- Windows 7(64bit)
- Visual Studio 2013
- Caffe
- MNIST
설치과정은 생략하고 데스트 위주로 정리합니다.
LeNet을 활용한 MNIST 예제
- Training과 Test에 사용할 데이터 다운로드
< 참고파일 : .\data\mnist\get_mnist.sh >
(caffe가 설치된 폴더를 기준으로 작성하였습니다.)
* 아래 링크들을 클릭하시면 다운로드 하실 수 있습니다.
* 압축을 해제하시면 아래의 파일들이 있습니다.
(이 파일들을 '.\data\mnist\'로 옮겨 주세요, 이후에 경로 지정을 이곳으로 합니다.)
train-images.idx3-ubyte
train-labels.idx1-ubyte
t10k-images.idx3-ubyte
t10k-labels.idx1-ubyte
- lmdb로 변환
< 참고파일 : .\examples\mnist\create_mnist.sh >
* convert_minist_data.exe을 활용하여 lmdb로 변환합니다.
첫 번째 training 데이터 입니다.
> .\Build\x64\Debug\convert_mnist_data.exe .\data\mnist\train-images.idx3-ubyte .\data\mnist\train-labels.idx1-ubyte .\examples\mnist\mnist_train_lmdb --backend=lmdb
두 번째로는 test 데이터 입니다.
> .\Build\x64\Debug\convert_mnist_data.exe .\data\mnist\t10k-images.idx3-ubyte .\mnist\t10k-labels.idx1-ubyte .\examples\mnist\mnist_test_lmdb --backend=lmdb
결과는 아래의 그림과 같이 새로 만들어진 폴더를 확인 하실 수 있습니다.
- Net과 Solver는 caffe에서 제공해주는 것을 그대로 사용하겠습니다.
* Net 위치 : .\examples\mnist\lenet_train_test.prototxt
* Solver 위치 : .\examples\mnist\lenet_solver.prototxt
<TroubleShooting>
lenet_train_test.prototxt 파일에 보시면 아래 그림과 같이 레이어 안에 'source' 위치를 지정하는 곳이 있습니다. 이 곳은 이전 "lmdb로 변환"에서 결과를 저장하는 곳 ("training : .\examples\mnist\mnist_train_lmdb"과 "test : .\examples\mnist\mnist_test_lmdb"로 각각 지정한 곳입니다.)
- Training 과정
< 참고파일 : .\examples\mnist\train_lenet.sh >
> .\Builtx64\Debug\caffe.exe train --solver=examples\mnist\lenet_solver_prototxt
결과로 생성된 파일은 아래의 그림과 같습니다.
'STUDY > AI' 카테고리의 다른 글
A Step by Step Backpropagation Example (Backpropagation 예제를 활용한 한걸음) (0) | 2016.10.12 |
---|