[.m5 to .json] 텐서플로우 js .m5확장자 .json변경

Mnist로 만든 예제 모델이 tensorflow.js에서 잘 돌아가지 않았다. 확장자 명이 .m5였는데, 다른 사람들의 예제를 보니 모델 확장자 명이 .json이었다.

 

 터미널에 아래 명령어를 입력하여 .m5파일을 .json으로 변경 가능하다.

tensorflowjs_converter --input_format=keras /tmp/model.h5 /tmp/tfjs_model

 

혹은 파이썬으로 아래 코드를 입력하고 파이썬을 실행시킨다.

from keras.models import load_model

 

model = load_model('my_model.h5')

 

model_json = model.to_json()

 

with open("model.json", "w") as json_file:

    json_file.write(model_json)

 

그럼 model.json이라는 이름으로 json파일이 생성된다.