keras load_model함수 버전 오류

Raspberry에서 keras.models.load_model 혹은 tensofrlos.keras.models.load_model을 하니 오류가 생겼다.

원인은 Google Colab에서 만든 model을 load_model하려니 tensorflow의 버전이 안맞는 것이었다.

Colab은 tensorflow 2.3을 사용하고 라즈베리에서 pip3 install tensorflow를 하면 2.0이 설치된다.

따라서 해결방법은 다음과 같다.

 

1. tensorflow 2.2 설치하기

github.com/lhelontra/tensorflow-on-arm/releases

Releases · lhelontra/tensorflow-on-arm

위 주소로 이동해서 wget으로 whl을 받은 다음 pip3 install로 설치하는 것이다. 2.3을 바로 설치하면 좋겠지만 이상하게 2.3은 설치가 안되서 우선 2.2로 설치하였다.

 

 

2. load_model에 옵션주기

2.2로 하면 tensorflow 내부에서 또다른 오류 메시지가 생긴다. 따라서 다음과 같이 load_model을 사용하자

keras.models.load_model('saved_model', compile=False)