[NODEJS] tensorflow 모델 업로드

다음 사진은 tensorflowjs coverter로 변환한 후 저장된 모습이다.

모델을 웹상으로 업로드 하는 코드는 다음과 같다.

Node.js로 실행할 때 파일 시스템에 직접 액세스 하는 방법으로 모델을 저장할 수 있다.

await model.save('file:///path/to/my-model');

model을 로드하는 방법으로 tf.loadLayersModel api를 사용하였다.

const model = await tf.loadLayersModel('localstorage://my-model-1');

완성된 코드는 다음과 같다.

await 문법을 사용하기 위해서는 async 예약어를 사용해주어야 한다고 한다.

const express = require('express');
const tf = require('@tensorflow/tfjs-node');

// 생략

async function loadModel() {
    const model = await tf.loadLayersModel('./tfjs_model');
    await model.save('file://./tfjs_model');
}

결과 화면은 다음과 같다.

 

참고자료

https://www.tensorflow.org/js/guide/save_load?hl=ko 

모델 저장하기 및 로드하기  |  TensorFlow.js