딥러닝

Mnist 학습 후 c#으로 실행하기

카멜레온개발자 2020. 3. 10. 10:30

* 먼저 PC에 python, tensorflow GPU 버전, Cuda, cudnn이 설치되어 있어야 함

 나의 PC

 - Python : 3.7

 - Tensorflow : 1.15 (GPU버전)

 - cuda : 10.0

 - cudnn : 7.x (찾기 귀찮음...-_-)

 - Visual Studio 2017

 

1. 아래의 위치에서 다운받아 적당한 위치에 압축 해제를 한다.

 - https://github.com/sdg002/MNISTpng

 

2. training.zip, testing.zip 파일을 압축 해제 한다.

 * [소스폴더]\MNISTpng\training.zip -> [소스폴더]\MNISTpng\training\0 ~ 9

 * [소스폴더]\MNISTpng\testing.zip -> [소스폴더]\MNISTpng\testing\0 ~ 9

 

3. [소스폴더]\PythonTrainer\TrainMnistFromFolder.py 파일을 다음과 같이 수정한다.

 - mnist_train_path_full="..\\MNISTpng\\training\\*\\*.png"

 - mnist_test_path="..\\MNISTpng\\testing\\*\\*.png"

 

4. 파이썬으로 빌드해준다. (GPU 버전이 아니면 안되는듯)

 

5. 정상적으로 완료되면 [소스폴더]\PythonTrainer\Out 폴더에 Mnist_model.pb파일이 생성된다.

 

6. [소스폴더]에 MNISTpng.sln 파일을 Visual Stuido 2017로 오픈한다.

 

7. 솔루션 탐색기에서 ConsoleAppTester 프로젝트의 Program.cs 파일을 열어 다음과 같이 수정한 다음,

   ConsoleAppTester 프로젝트를 빌드한다.

 

 

8. 실행하려고 하면 실패하는데, 실행화일이 있는 디버그 폴더로 가 보면 libtensorflow.zip 파일이 있는데, 그 파일을 현재 폴더로 압축 해제 한다.

 * 압축 해제 하면 libtensorflow.dll 파일이 나온다.

 

9. 이제 실행해 보면 실행이 되어야 한다.