TensorFlow 1.x vs 2.x버전 MNIST학습 데이터 import 차이

안녕하세요

구글에서 제공해주는 Colab(Colaboratory)을 이용해 MNIST코딩을 해보는 중

TensorFlow 버전 별 MNIST import 방식이 다른 듯 하여 포스팅해놓습니다.

 

결론적으로

TensorFlow 2.0 버전의 튜토리얼이 자세히 나와있어 2.0 버전으로 공부하는 것이 괜찮을 것 같습니다.

1. Tensorflow 2.x 에서 Tensorflow 1.x 방식으로 MNIST import 오류

TensorFlow 1.x 에서는 아래 패키지를 import하게 되어있습니다.

  -> from tensorflow.examples.tutorials.mnist import input_data

 

제가 산 책이 Tensorflow 1.x 버전이었는데 Colab TensorFlow가 2.2 버전 이었기에 아래와 같이 오류가 발생했습니다.

github issue를 검색해보니 2.x버전부터는 아래 MNIST 예제 패키지를 제공하지 않는 것 같습니다.

 

Tensorflow 2.x 방식에서의 MNIST import 오류

Github issue 의견들

출처 : https://github.com/tensorflow/tensorflow/issues/34083

2. MNIST 데이터 import 오류 해결

  1) TensorFlow 2.x 버전으로 설치한 뒤 tf.keras.datasets.mnist의 사용

      구글에서 제공하는 TensorFlow 2.0를 보면 tf.keras.datasets.mnist를 사용해 MNIST 데이터셋을 import합니다.

      아래 참조 튜토리얼 가이드대로 학습하면 됩니다.

참조 : https://www.tensorflow.org/tutorials/quickstart/beginner?hl=ko

텐서플로 2.0 시작하기: 초보자용  |  TensorFlow Core

  2) TensorFlow 1.x 버전으로 다운그레이드 한 뒤 from tensorflow.examples.tutorials.mnist import 사용

      Tensorflow 1.x 버전으로 다운그레이드 한 뒤

      from tensorflow.examples.tutorials.mnist import input_data를 통해 데이터셋을 import합니다.

      많은 책 및 블로그가 현재도 1.x 버전 기준으로 설명하고 있기에 간단하게 실습하기는 더 편한 것 같습니다.

 

  3) Local에 MNIST 데이터셋을 저장하여 import 사용

       버전과 관련 없이 Local폴더에 MNIST Dataset을 저장하여 활용합니다.